Skip to content

Commit 910053f

Browse files
authored
Fix ATen mode op_upsample_bilinear2d_test
Differential Revision: D68930414 Pull Request resolved: #8087
1 parent afc5a50 commit 910053f

File tree

1 file changed

+6
-0
lines changed

1 file changed

+6
-0
lines changed

kernels/test/op_upsample_bilinear2d_test.cpp

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -44,6 +44,12 @@ class OpUpsampleBilinear2dTest : public OperatorTest {
4444
void test_upsample_bilinear2d_dtype() {
4545
TensorFactory<DTYPE> tf;
4646

47+
if (torch::executor::testing::SupportedFeatures::get()->is_aten &&
48+
(DTYPE == ScalarType::Char || DTYPE == ScalarType::Short ||
49+
DTYPE == ScalarType::Int || DTYPE == ScalarType::Long)) {
50+
// not supported.
51+
return;
52+
}
4753
const auto input = tf.make({1, 1, 1, 2}, {1, 4});
4854
std::array<int64_t, 2> output_size = {1, 4};
4955
auto out = tf.zeros({1, 1, 1, 4});

0 commit comments

Comments
 (0)