f

microsoft · Oct 19, 2024 · fec9d5a · fec9d5a
1 parent db29664
commit fec9d5a
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/onnxruntime/test/contrib_ops/layer_norm_op_test.cc b/onnxruntime/test/contrib_ops/layer_norm_op_test.cc
@@ -280,7 +280,7 @@ TYPED_TEST(LayerNormTest, LayerNorm17_opset) {
     test.AddInput<TypeParam>("x", dims, GetTypedArray<TypeParam>({1.0f, 2.0f, 3.0f, 4.0f, 5.0f, 6.0f}));
     test.AddInput<TypeParam>("gamma", {3}, GetTypedArray<TypeParam>({1.0f, 1.0f, 1.0f}), is_initializer);
     test.AddOutput<TypeParam>("output", dims, GetTypedArray<TypeParam>({-1.2247f, 0.0f, 1.2247f, -1.2247f, 0.0f, 1.2247f}));
-    if (std::is_same<TypeParam, MLFloat16>::value|| is_initializer) {
+    if (std::is_same<TypeParam, MLFloat16>::value) {
       std::vector<std::unique_ptr<IExecutionProvider>> execution_providers;
       execution_providers.push_back(DefaultCoreMLExecutionProvider(true));
       // coreml EP requires weight and bias to be initializers
@@ -292,8 +292,14 @@ TYPED_TEST(LayerNormTest, LayerNorm17_opset) {
       test.Run();
     }
   };
+  if (std::is_same<TypeParam, MLFloat16>::value) {
+#if !definded(COREML_ENABLE_MLPROGRAM)  && !defined(USE_CUDA)
+  return;
+#endif
+  }
+
   run_test(false);
-#ifdef COREML_ENABLE_MLPROGRAM
+#if definded(COREML_ENABLE_MLPROGRAM)
   // gamma as initialized will fail for CPUEP
   run_test(true);
 #endif