Unverified Commit 5ee93a59 authored by Zhewen Li's avatar Zhewen Li Committed by GitHub
Browse files

[CI/Build] Update checking logic in cutlass_group_gemm_supported (#27948)


Signed-off-by: default avatarzhewenli <zhewenli@meta.com>
parent e1560178
...@@ -836,7 +836,11 @@ def cutlass_sparse_scaled_mm_supported(cuda_device_capability: int) -> bool: ...@@ -836,7 +836,11 @@ def cutlass_sparse_scaled_mm_supported(cuda_device_capability: int) -> bool:
def cutlass_group_gemm_supported(cuda_device_capability: int) -> bool: def cutlass_group_gemm_supported(cuda_device_capability: int) -> bool:
try:
return torch.ops._C.cutlass_group_gemm_supported(cuda_device_capability) return torch.ops._C.cutlass_group_gemm_supported(cuda_device_capability)
except AttributeError:
# Return False on non-CUDA platforms where it is not available
return False
def cutlass_sparse_compress(a: torch.Tensor) -> tuple[torch.Tensor, torch.Tensor]: def cutlass_sparse_compress(a: torch.Tensor) -> tuple[torch.Tensor, torch.Tensor]:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment