tests/jax/test_custom_call_compute.py · 210e57de151cd19f69ff6ae7b4a8363eae8489eb · OpenDAS / TransformerEngine

Kirthi Shankar Sivamani authored Jun 13, 2024



* Apply formatting
Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>

* Apply formatting
Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>

---------
Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>

9416519d

test_custom_call_compute.py 28.3 KB

Replace test_custom_call_compute.py