|
2 | 2 | // Licensed under the MIT License. |
3 | 3 |
|
4 | 4 | #include "gtest/gtest.h" |
| 5 | +#include "test/common/cuda_op_test_utils.h" |
5 | 6 | #include "test/providers/provider_test_utils.h" |
6 | 7 |
|
7 | 8 | #include <limits> |
@@ -125,20 +126,32 @@ TEST(IsInfTest, test_isinf_negative_mlfloat16) { |
125 | 126 | } |
126 | 127 |
|
127 | 128 | TEST(IsInfTest, test_isinf_bfloat16) { |
| 129 | + if (NeedSkipIfCudaArchLowerThan(800)) { |
| 130 | + GTEST_SKIP() << "Skipping BFloat16 tests on CUDA Compute Capability < 8.0"; |
| 131 | + } |
| 132 | + |
128 | 133 | std::initializer_list<BFloat16> input = {BFloat16{-1.7f}, BFloat16::NaN, BFloat16::Infinity, 3.6_bfp16, |
129 | 134 | BFloat16::NegativeInfinity, BFloat16::Infinity}; |
130 | 135 | std::initializer_list<bool> output = {false, false, true, false, true, true}; |
131 | | - run_is_inf_test(20, 1, 1, input, output, true); // Skip as TRT10 supports BF16 but T4 GPU run on TRT CIs doesn't |
| 136 | + run_is_inf_test(20, 1, 1, input, output); |
132 | 137 | } |
133 | 138 |
|
134 | 139 | TEST(IsInfTest, test_isinf_positive_bfloat16) { |
| 140 | + if (NeedSkipIfCudaArchLowerThan(800)) { |
| 141 | + GTEST_SKIP() << "Skipping BFloat16 tests on CUDA Compute Capability < 8.0"; |
| 142 | + } |
| 143 | + |
135 | 144 | std::initializer_list<BFloat16> input = {BFloat16{-1.7f}, BFloat16::NaN, BFloat16::Infinity, 3.6_bfp16, |
136 | 145 | BFloat16::NegativeInfinity, BFloat16::Infinity}; |
137 | 146 | std::initializer_list<bool> output = {false, false, true, false, false, true}; |
138 | 147 | run_is_inf_test(20, 1, 0, input, output); |
139 | 148 | } |
140 | 149 |
|
141 | 150 | TEST(IsInfTest, test_isinf_negative_bfloat16) { |
| 151 | + if (NeedSkipIfCudaArchLowerThan(800)) { |
| 152 | + GTEST_SKIP() << "Skipping BFloat16 tests on CUDA Compute Capability < 8.0"; |
| 153 | + } |
| 154 | + |
142 | 155 | std::initializer_list<BFloat16> input = {BFloat16{-1.7f}, BFloat16::NaN, BFloat16::Infinity, 3.6_bfp16, |
143 | 156 | BFloat16::NegativeInfinity, BFloat16::Infinity}; |
144 | 157 | std::initializer_list<bool> output = {false, false, false, false, true, false}; |
|
0 commit comments