diff --git a/onnxruntime/test/contrib_ops/element_wise_ops_test.cc b/onnxruntime/test/contrib_ops/element_wise_ops_test.cc index a7debe845e..03a67b539b 100644 --- a/onnxruntime/test/contrib_ops/element_wise_ops_test.cc +++ b/onnxruntime/test/contrib_ops/element_wise_ops_test.cc @@ -89,18 +89,16 @@ static void RunBiasGeluTest( const std::vector& input_b_data, const std::vector& input_a_dims, const std::vector& input_b_dims) { - if (HasCudaEnvironment(0)) { - std::vector output_data = ComputeGeluWithErf(Add_Simple(input_a_data, input_b_data)); + std::vector output_data = ComputeGeluWithErf(Add_Simple(input_a_data, input_b_data)); - OpTester tester("BiasGelu", 1, onnxruntime::kMSDomain); + OpTester tester("BiasGelu", 1, onnxruntime::kMSDomain); - const std::vector& output_dims = input_a_dims.size() >= input_b_dims.size() ? input_a_dims : input_b_dims; - tester.AddInput("A", input_a_dims, input_a_data); - tester.AddInput("B", input_b_dims, input_b_data); - tester.AddOutput("C", output_dims, output_data); + const std::vector& output_dims = input_a_dims.size() >= input_b_dims.size() ? input_a_dims : input_b_dims; + tester.AddInput("A", input_a_dims, input_a_data); + tester.AddInput("B", input_b_dims, input_b_data); + tester.AddOutput("C", output_dims, output_data); - tester.Run(); - } + tester.Run(); } TEST(BiasGeluTest, Two_One_Dim) {