Delete deprecated BlockSparseLayout and related code

[jerryzh168]

Stack from ghstack (oldest at bottom):

• Delete deprecated TensorCoreTiledLayout and related code #4153
• Move bitpacking.py to prototype and add uintx_utils.py #4152
• Delete deprecated PlainLayout, PlainAQTTensorImpl and related v1 code paths #4151
• Delete deprecated PackedLinearInt8DynamicActivationIntxWeightLayout and related code #4150
• Delete deprecated Int4CPULayout and related code #4149
• Delete deprecated SemiSparseLayout and related code #4148
• Delete deprecated Int4XPULayout and related code #4147
• Delete deprecated QDQLayout and related code #4146
• Delete deprecated UintxLayout and related code #4145
• Delete deprecated GemlitePackedLayout and GemliteUIntXWeightOnlyConfig #4144
• -> Delete deprecated BlockSparseLayout and related code #4143

[pytorch-bot]

🔗 Helpful Links

🧪 See artifacts and rendered test results at hud.pytorch.org/pr/pytorch/ao/4143

• 📄 Preview Python docs built from this PR

Note: Links to docs will display an error until the docs builds have been completed.

✅ You can merge normally! (1 Unrelated Failure)

As of commit b709232 with merge base 34322b5 ():

BROKEN TRUNK - The following job failed but were present on the merge base:

👉 Rebase onto the `viable/strict` branch to avoid these failures

• Run Regression Tests / test-nightly (CUDA Nightly, linux.g5.12xlarge.nvidia.gpu, --pre torch --index-url https://downloa... / linux-job (gh) (trunk failure)
  test/test_low_bit_optim.py::TestFSDP2::test_fsdp2

This comment was automatically generated by Dr. CI and updates every 15 minutes.

[jerryzh168]

@claude review

[claude]

Claude finished @jerryzh168's task in 1m 33s —— View job

---

Review of PR #4143: Delete deprecated BlockSparseLayout and related code

• Gather context and read changed files
• Analyze the diff for correctness and completeness
• Post review feedback

---

Overall: This PR looks good. It cleanly removes the deprecated BlockSparseLayout and BlockSparseAQTTensorImpl code along with the associated test, deprecation shim, and dispatch registration. The deletion is consistent and well-scoped.

Summary of changes

• Deleted torchao/prototype/dtypes/uintx/block_sparse_layout.py (the actual implementation, 233 lines)
• Deleted torchao/dtypes/uintx/block_sparse_layout.py (the deprecation stub, 24 lines)
• Removed BlockSparseLayout from __init__.py exports in torchao/dtypes/, torchao/prototype/dtypes/, and torchao/prototype/dtypes/uintx/
• Removed the block sparse check/impl dispatch registration from affine_quantized_tensor_ops.py
• Removed the TestQuantBlockSparseWeight test class from test/sparsity/test_sparse_api.py

Issue found

Stale deprecation test reference — test/dtypes/test_uintx.py:105 still references the now-deleted module:

deprecated_apis = [
    ("BlockSparseLayout", "torchao.dtypes.uintx.block_sparse_layout"),  # line 105
    ("UintxLayout", "torchao.dtypes.uintx.uintx_layout"),
]

This test (test_uintx_api_deprecation) will fail at runtime because:

1. The module torchao.dtypes.uintx.block_sparse_layout no longer exists
2. The exec(f"from torchao.dtypes import BlockSparseLayout") call will fail since BlockSparseLayout was removed from torchao/dtypes/__init__.py

This entry should be removed from the deprecated_apis list.

Fix this →

Notes

• The non-quantized TestBlockSparseWeight test class and block_sparse_weight sparsification API in torchao/sparsity/ are correctly left intact — this PR only removes the quantized block sparse layout path.
• No other files in the codebase reference BlockSparseLayout, BlockSparseAQTTensorImpl, or block_sparse_layout beyond the stale test noted above.

---

[jerryzh168]

@claude review again

[claude]

Claude finished @jerryzh168's task in 1m 1s —— View job

---

Re-review of PR #4143: Delete deprecated BlockSparseLayout and related code

• Gather context and read changed files
• Check if previously flagged issue was addressed
• Analyze the full diff for correctness and completeness
• Post review feedback

---

Overall: This PR looks good. The previously flagged issue (stale BlockSparseLayout entry in the deprecation test at test/dtypes/test_uintx.py) has been fixed in the updated commit.

Summary of changes (8 files, +0/-318)

• Deleted torchao/prototype/dtypes/uintx/block_sparse_layout.py — the 233-line implementation (BlockSparseLayout, BlockSparseAQTTensorImpl, and the int8 linear dispatch functions)
• Deleted torchao/dtypes/uintx/block_sparse_layout.py — the 24-line deprecation shim
• Removed BlockSparseLayout from __init__.py exports in torchao/dtypes/, torchao/prototype/dtypes/, and torchao/prototype/dtypes/uintx/
• Removed the block sparse check/impl dispatch registration from torchao/dtypes/affine_quantized_tensor_ops.py
• Removed TestQuantBlockSparseWeight test class and its instantiate_parametrized_tests call from test/sparsity/test_sparse_api.py
• Removed the BlockSparseLayout entry from the deprecation test in test/dtypes/test_uintx.py

Verification

• Searched the entire codebase for BlockSparseLayout, BlockSparseAQTTensorImpl, and block_sparse_layout — zero remaining references. The cleanup is complete.
• The non-quantized TestBlockSparseWeight test class and block_sparse_weight sparsification API in torchao/sparsity/ are correctly left intact.
• CI: Ruff, Build Docs, and PR Label Check are passing. GPU/regression tests are still in progress.

No issues found.

---

View job