graph: backend: dnnl: fixup f16 data type check on gpu

oneapi-src · Apr 27, 2024 · fdff2ef · fdff2ef
1 parent d50fa18
commit fdff2ef
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/graph/backend/dnnl/platform.cpp b/src/graph/backend/dnnl/platform.cpp
@@ -76,8 +76,8 @@ bool get_dtype_support_status(engine_kind_t eng, data_type_t dtype, dir_t dir) {
             break;
         }
         case dnnl_f16: {
-            // f16 is supported on GPU for inference only.
-            is_supported = is_gpu(eng) && (dir & dir_t::FLAG_FWD);
+            // TODO(zhitao): f16 for backward on gpu?
+            is_supported = is_gpu(eng);
             break;
         }
         case dnnl_f8_e5m2: {