George Necula 0831e2e340 [shape_poly] Adding shape polymorphism support for the state primitives.

2024-11-21 06:17:01 -08:00

..

__init__.py

Cleanup: fix unused imports & mark exported names

2024-10-16 17:42:41 -07:00

discharge.py

Make trace dispatch purely a function of context rather than a function of both context and data. This lets us delete a lot of machinery for managing data-dependent tracing: levels, sublevels, post_process_call, new_base_main, custom_bind and so on.

2024-10-29 11:04:31 -07:00

indexing.py

[shape_poly] Adding shape polymorphism support for the state primitives.

2024-11-21 06:17:01 -08:00

primitives.py

Reverts f281c6f46475270a57a02416469226315377592c

2024-11-05 07:17:14 -08:00

types.py

Reverts f281c6f46475270a57a02416469226315377592c

2024-11-05 07:17:14 -08:00

utils.py

[Pallas TPU] Refactor ref indexers to transforms and support ref bitcast.

2024-09-15 17:53:29 -07:00