Add float4_e2m1fn_x2 to lcdtype_to_nvdtype_map (#2532)

crcrpar · web-flow · commit 57400020fa30 · 2025-10-09T13:30:05.000+02:00
diff --git a/thunder/executors/nvfuserex_impl.py b/thunder/executors/nvfuserex_impl.py
@@ -122,6 +122,14 @@
         }
     )
 
+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,
+        }
+    )
+
 _lcfp8_to_nvfp8_map: dict[dtypes.dtype, DataType] = {
     dtypes.float8_e5m2: DataType.Float8_e5m2,
     dtypes.float8_e5m2_: DataType.Float8_e5m2,

Original file line number	Diff line number	Diff line change
`@@ -122,6 +122,14 @@`
`122`	`122`	`}`
`123`	`123`	`)`
`124`	`124`
	`125`	`+if nvfuser_version() >= LooseVersion("0.2.28"):`
	`126`	`+ _lcdtype_to_nvdtype_map.update(`
	`127`	`+ {`
	`128`	`+ dtypes.float4_e2m1fn_x2: DataType.Float4_e2m1fn_x2,`
	`129`	`+ dtypes.float4_e2m1fn_x2_: DataType.Float4_e2m1fn_x2,`
	`130`	`+ }`
	`131`	`+ )`
	`132`	`+`
`125`	`133`	`_lcfp8_to_nvfp8_map: dict[dtypes.dtype, DataType] = {`
`126`	`134`	`dtypes.float8_e5m2: DataType.Float8_e5m2,`
`127`	`135`	`dtypes.float8_e5m2_: DataType.Float8_e5m2,`