fp4 since 0.2.28

crcrpar · crcrpar · commit 8b6a38da19a8 · 2025-10-07T14:48:52.000+09:00
Signed-off-by: Masaki Kozuki &lt;mkozuki@nvidia.com&gt;
diff --git a/thunder/executors/nvfuserex_impl.py b/thunder/executors/nvfuserex_impl.py
@@ -119,6 +119,12 @@
         {
             dtypes.uint64: DataType.UInt64,
             dtypes.uint64_: DataType.UInt64,
+        }
+    )
+
+if nvfuser_version() >= LooseVersion("0.2.28"):
+    _lcdtype_to_nvdtype_map.update(
+        {
             dtypes.float4_e2m1fn_x2: DataType.Float4_e2m1fn_x2,
             dtypes.float4_e2m1fn_x2_: DataType.Float4_e2m1fn_x2,
         }

Original file line number	Diff line number	Diff line change
`@@ -119,6 +119,12 @@`
`119`	`119`	`{`
`120`	`120`	`dtypes.uint64: DataType.UInt64,`
`121`	`121`	`dtypes.uint64_: DataType.UInt64,`
	`122`	`+ }`
	`123`	`+ )`
	`124`	`+`
	`125`	`+if nvfuser_version() >= LooseVersion("0.2.28"):`
	`126`	`+ _lcdtype_to_nvdtype_map.update(`
	`127`	`+ {`
`122`	`128`	`dtypes.float4_e2m1fn_x2: DataType.Float4_e2m1fn_x2,`
`123`	`129`	`dtypes.float4_e2m1fn_x2_: DataType.Float4_e2m1fn_x2,`
`124`	`130`	`}`