trunk/160ef925057d994a806e29a078ded765a03d64c5: [ROCm] Enable test_cslt_sparse_mm_alpha for fp16 and bf16 (#199050)
- PyTorch 近 90 天出现 1780 次
- PyTorch 近 90 天第 1758 次发版
- 上一次:同一天稍早 · ciflow/trunk/199110
发生了什么
Summary Fixes #168851 Fixes #168852 This PR enables test_cslt_sparse_mm_alpha for float16 and bfloat16. The test passes now has both cuSPARSELt and hipSparseLt applying bias epilogue. The dense reference was only alpha * (A @ B), so it was off by 1 on every element. Add the bias to the reference on every platform and drop the CUDA skip. That skip called out cuSPARSELt v0.6.x, which CI no longer ships. Test plan pyth…
摘要按规则整理自下方来源原文