Skip to content
Merged
Show file tree
Hide file tree
Changes from 1 commit
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 2 additions & 0 deletions onnxruntime/test/providers/cpu/nn/conv_fp16_test.cc
Original file line number Diff line number Diff line change
Expand Up @@ -282,6 +282,7 @@ TEST(ConvBF16Test, Conv2D_1) {
return;
#endif

#ifdef USE_CUDA
Comment thread
hariharans29 marked this conversation as resolved.
Outdated
OpTester test("Conv", 22);

ConvOpAndTestAttributes attributes = {
Expand Down Expand Up @@ -332,6 +333,7 @@ TEST(ConvBF16Test, Conv2D_1) {
test.AddOutput<BFloat16>("Y", Y_shape, expected_vals, /*no sort*/ false, 0.002f, 0.0f);

test.Run(OpTester::ExpectResult::kExpectSuccess, "", {kCudaExecutionProvider});
#endif
}

TEST(ConvFp16Test, Conv2D_1) {
Expand Down
2 changes: 2 additions & 0 deletions onnxruntime/test/providers/cpu/nn/pool_fp16_op_test.cc
Original file line number Diff line number Diff line change
Expand Up @@ -332,6 +332,7 @@ TEST(PoolBF16Test, AveragePool) {
return;
#endif

#ifdef USE_CUDA
OpTester test("AveragePool", 22);

test.AddAttribute("auto_pad", "");
Expand Down Expand Up @@ -411,6 +412,7 @@ TEST(PoolBF16Test, AveragePool) {
test.AddInput<BFloat16>("X", x_dims, x_vals);
test.AddOutput<BFloat16>("Y", expected_dims, expected_vals);
test.Run(OpTester::ExpectResult::kExpectSuccess, "", {kCudaExecutionProvider});
#endif
}

TEST(PoolFp16Test, AveragePool) {
Expand Down
Loading