mirror of https://github.com/llvm/torch-mlir
c77f3b559a
While waiting for the full resolution of feature request https://github.com/pytorch/pytorch/issues/117188 (which will propagate sparsity the right way in upstream PyTorch for all FX Graphs), this minor change allows us to start testing sparsity "within" a network, rather than just the parameters. Feel free to add your own rules for testing (but within reason for what will be done upstream). Note, two TODOs need to be addressed to work around some pending issues to make the JIT execution work. |
||
|---|---|---|
| .. | ||
| v2.3 | ||
| basic_test.py | ||
| sparse_test.py | ||