Skip to content

Commit 0f436dc

Browse files
committed
Fix ATen mode op_upsample_bilinear2d_test
Was broken, now it's not. Differential Revision: [D68930414](https://our.internmc.facebook.com/intern/diff/D68930414/) ghstack-source-id: 263966264 Pull Request resolved: #8087
1 parent afc5a50 commit 0f436dc

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)