saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-07-19 19:00:47 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Ensure that the BiasGelu op test runs on CPU. (#8671)

Browse source
This commit is contained in:
Nick Kreeger 2021-08-10 14:59:41 -05:00 committed by GitHub
parent e74f86059c
commit 3a742f2910
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 7 additions and 9 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

16
onnxruntime/test/contrib_ops/element_wise_ops_test.cc
View file

@ -89,18 +89,16 @@ static void RunBiasGeluTest(
const std::vector<float>& input_b_data,
const std::vector<int64_t>& input_a_dims,
const std::vector<int64_t>& input_b_dims) {
if (HasCudaEnvironment(0)) {
std::vector<float> output_data = ComputeGeluWithErf(Add_Simple(input_a_data, input_b_data));
std::vector<float> 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<int64_t>& output_dims = input_a_dims.size() >= input_b_dims.size() ? input_a_dims : input_b_dims;
tester.AddInput<float>("A", input_a_dims, input_a_data);
tester.AddInput<float>("B", input_b_dims, input_b_data);
tester.AddOutput<float>("C", output_dims, output_data);
const std::vector<int64_t>& output_dims = input_a_dims.size() >= input_b_dims.size() ? input_a_dims : input_b_dims;
tester.AddInput<float>("A", input_a_dims, input_a_data);
tester.AddInput<float>("B", input_b_dims, input_b_data);
tester.AddOutput<float>("C", output_dims, output_data);
tester.Run();
}
tester.Run();
}
TEST(BiasGeluTest, Two_One_Dim) {