fix: blackwell dsv3 fp8 issue temporary solution (#9530)

This commit is contained in:
Yineng Zhang
2025-08-23 01:33:21 -07:00
committed by GitHub
parent 4edbe0d534
commit 6b2b8bf0e1

View File

@@ -7,7 +7,8 @@ logger = logging.getLogger(__name__)
def _compute_enable_deep_gemm():
sm_version = get_device_sm()
if sm_version < 90:
# TODO fix blackwell fp8
if sm_version != 90:
return False
try: