add float4_e2m1fn_x2 to lcdtype_to_nvdtype_map

crcrpar · crcrpar · commit 40b04ba6e674 · 2025-09-25T17:07:24.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,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
`@@ -114,6 +114,8 @@`
`114`	`114`	`{`
`115`	`115`	`dtypes.uint64: DataType.UInt64,`
`116`	`116`	`dtypes.uint64_: DataType.UInt64,`
	`117`	`+ dtypes.float4_e2m1fn_x2: DataType.Float4_e2m1fn_x2,`
	`118`	`+ dtypes.float4_e2m1fn_x2_: DataType.Float4_e2m1fn_x2,`
`117`	`119`	`}`
`118`	`120`	`)`
`119`	`121`