fix mapping in common.h

phu0ngng · phu0ngng · commit a8ce721862d1 · 2025-11-24T18:25:21.000-08:00
Signed-off-by: Phuong Nguyen &lt;phuonguyen@nvidia.com&gt;
diff --git a/transformer_engine/common/common.h b/transformer_engine/common/common.h
@@ -501,7 +501,7 @@ struct TypeInfo {
       { __VA_ARGS__ }                                        \
     } break;                                                 \
     case DType::kFloat64: {                                  \
-      using type = float;                                    \
+      using type = double;                                   \
       { __VA_ARGS__ }                                        \
     } break;                                                 \
     case DType::kFloat16: {                                  \
