add float4_e2m1fn_x2 to lcdtype_to_nvdtype_map

crcrpar · crcrpar · commit c8006f883c78 · 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,8 @@
         {
             dtypes.uint64: DataType.UInt64,
             dtypes.uint64_: DataType.UInt64,
+            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,8 @@`
`119`	`119`	`{`
`120`	`120`	`dtypes.uint64: DataType.UInt64,`
`121`	`121`	`dtypes.uint64_: DataType.UInt64,`
	`122`	`+ dtypes.float4_e2m1fn_x2: DataType.Float4_e2m1fn_x2,`
	`123`	`+ dtypes.float4_e2m1fn_x2_: DataType.Float4_e2m1fn_x2,`
`122`	`124`	`}`
`123`	`125`	`)`
`124`	`126`