We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent d4297d5 commit 4177660Copy full SHA for 4177660
src/aiconfigurator/sdk/task.py
@@ -542,6 +542,11 @@ def _get_quant_mode(
542
comm_quant_mode = "half"
543
544
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
550
551
supported = getattr(database, "supported_quant_mode", {}) or {}
552
supported_gemm = set(supported.get("gemm", []) or [])
0 commit comments