Skip to content

Commit 4177660

Browse files
committed
add sm vesion comment
1 parent d4297d5 commit 4177660

File tree

1 file changed

+5
-0
lines changed

1 file changed

+5
-0
lines changed

src/aiconfigurator/sdk/task.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -542,6 +542,11 @@ def _get_quant_mode(
542542
comm_quant_mode = "half"
543543

544544
sm_version = database.system_spec["gpu"]["sm_version"]
545+
# SM version to GPU type mapping:
546+
# SM 80 = A100
547+
# SM 89 = L40S (Ada Lovelace) - no TMA, limited FP8 support
548+
# SM 90 = H100/H200 (Hopper) - full FP8 and TMA support
549+
# SM 100 = B200/GB200 (Blackwell) - NVFP4 support
545550

546551
supported = getattr(database, "supported_quant_mode", {}) or {}
547552
supported_gemm = set(supported.get("gemm", []) or [])

0 commit comments

Comments
 (0)