update in third_party xla

ROCm · Oct 1, 2024 · 18fc45a · 18fc45a
1 parent 069a82c
commit 18fc45a
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/third_party/xla/xla/tsl/framework/type_traits.h b/third_party/xla/xla/tsl/framework/type_traits.h
@@ -72,7 +72,9 @@ struct is_simple_type {
       is_quantized<T>::value || std::is_same<T, bfloat16>::value ||
       std::is_same<T, float8_e4m3fn>::value ||
       std::is_same<T, float8_e4m3b11fnuz>::value ||
-      std::is_same<T, float8_e5m2>::value || std::is_same<T, int4>::value ||
+      std::is_same<T, float8_e5m2>::value ||
+      std::is_same<T, float8_e4m3fnuz>::value ||
+      std::is_same<T, float8_e5m2fnuz>::value || std::is_same<T, int4>::value ||
       std::is_same<T, uint4>::value;
 };