fix: blackwell dsv3 fp8 issue temporary solution (#9530)
This commit is contained in:
@@ -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:
|
||||
|
||||
Reference in New Issue
Block a user