fp4 since 0.2.28

crcrpar · crcrpar · commit c0f7bf3d4f5e · 2025-09-26T03:28:04.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
@@ -114,6 +114,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
`@@ -114,6 +114,12 @@`
`114`	`114`	`{`
`115`	`115`	`dtypes.uint64: DataType.UInt64,`
`116`	`116`	`dtypes.uint64_: DataType.UInt64,`
	`117`	`+ }`
	`118`	`+ )`
	`119`	`+`
	`120`	`+if nvfuser_version() >= LooseVersion("0.2.28"):`
	`121`	`+ _lcdtype_to_nvdtype_map.update(`
	`122`	`+ {`
`117`	`123`	`dtypes.float4_e2m1fn_x2: DataType.Float4_e2m1fn_x2,`
`118`	`124`	`dtypes.float4_e2m1fn_x2_: DataType.Float4_e2m1fn_x2,`
`119`	`125`	`}`