torch-mlir/test/python/fx_importer
Aart Bik 24c2fc0b5f
[torch-mlir][sparse] add JIT test to expose pending issues (#2906)
This test exposes issues that need fixing
(1) propagate sparsity into the FX graph (over elt-wise) (2) batched
dimensions need a new "dense(batch)" format
2024-02-13 13:42:56 -08:00
..
basic_test.py Rename torch_mlir.compile APIs and introduce FX based analogs (#2842) 2024-02-06 19:07:59 -08:00
sparse_test.py [torch-mlir][sparse] add JIT test to expose pending issues (#2906) 2024-02-13 13:42:56 -08:00