From c8006f883c78ff88c99b9e92523b7e0df6f1fb0a Mon Sep 17 00:00:00 2001 From: Masaki Kozuki Date: Thu, 25 Sep 2025 17:07:24 +0900 Subject: [PATCH 1/2] add `float4_e2m1fn_x2` to lcdtype_to_nvdtype_map Signed-off-by: Masaki Kozuki --- thunder/executors/nvfuserex_impl.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/thunder/executors/nvfuserex_impl.py b/thunder/executors/nvfuserex_impl.py index f51e303380..20f688e3ca 100644 --- 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, } ) From 8b6a38da19a8739781a9efca9718bd5ba9fbecc7 Mon Sep 17 00:00:00 2001 From: Masaki Kozuki Date: Fri, 26 Sep 2025 03:28:04 +0900 Subject: [PATCH 2/2] fp4 since 0.2.28 Signed-off-by: Masaki Kozuki --- thunder/executors/nvfuserex_impl.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/thunder/executors/nvfuserex_impl.py b/thunder/executors/nvfuserex_impl.py index 20f688e3ca..4b6b59d588 100644 --- 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, }