Fix rowwise_scaled_linear_sparse_cutlass test#3768
Conversation
**Summary:** ``` pytest test/test_ops_rowwise_scaled_linear_sparse_cutlass.py ``` This gives ``` E RuntimeError: The size of tensor a (4) must match the size of tensor b (65) at non-singleton dimension 1 ``` Seems like it's caused by #3759. We just need to update the test. **Test Plan:** ``` pytest test/test_ops_rowwise_scaled_linear_sparse_cutlass.py ```
🔗 Helpful Links🧪 See artifacts and rendered test results at hud.pytorch.org/pr/pytorch/ao/3768
Note: Links to docs will display an error until the docs builds have been completed. ❗ 1 Active SEVsThere are 1 currently active SEVs. If your PR is affected, please view them below: ✅ No FailuresAs of commit 809ec3d with merge base 35fc480 ( This comment was automatically generated by Dr. CI and updates every 15 minutes. |
|
should we add the test run to CI? |
|
I think there are reasons why it wasn't added to CI, maybe because it needs special hardware or something? I'll let @jcaip do that when he gets back |
I think you can probably add to H100 test, but I'm also enabling the v2 version using this op in my PR any ways: https://github.com/pytorch/ao/pull/3727/changes#diff-bd400f440bccb67a4d23b5cc5c3ce843a1a6219354e25d69cd167827f62e415b |
Summary:
This gives
Seems like it's caused by #3759. We just need to update the test.
Test Plan: