Sharad Vikram 30973a9474 [Pallas] Pass in compiler params via explicit compiler_params argument instead of passing via **kwargs ...

This is a change that makes the API a bit more intuitive and avoids footguns like accidentally passing in `in_spec` instead of `in_specs` because previously kwargs that weren't used by any downstream lowering would be ignored and users would get weird errors as a result.

This change doesn't deprecate the old way of passing in compiler params but it will be deprecated soon after this.

PiperOrigin-RevId: 613239439

2024-03-06 09:16:22 -08:00

..

mosaic

[Pallas] Pass in compiler params via explicit compiler_params argument instead of passing via **kwargs

2024-03-06 09:16:22 -08:00

triton

[Pallas] Pass in compiler params via explicit compiler_params argument instead of passing via **kwargs

2024-03-06 09:16:22 -08:00

__init__.py

[Pallas] Upstream pallas to JAX

2023-08-01 16:43:13 -07:00

BUILD

Load Python rules from rules_python

2023-08-24 10:22:57 -07:00

core.py

Add support for non-disjoint windows in Pallas/Mosaic

2024-02-08 02:48:43 -08:00

pallas_call.py

[Pallas] Pass in compiler params via explicit compiler_params argument instead of passing via **kwargs

2024-03-06 09:16:22 -08:00

primitives.py

Add pl.num_programs to make it easier to query the dynamic grid size

2024-02-26 06:39:03 -08:00

utils.py

Reverts 8b1da123f84b37e54ed6d29b9a0822616df7f82a

2024-02-12 18:09:18 -08:00