From c45fb4bce76cefddb17366140d5719a425226571 Mon Sep 17 00:00:00 2001 From: Scott Wolchok Date: Thu, 30 Jan 2025 16:33:18 -0800 Subject: [PATCH] Fix ATen mode op_upsample_nearest2d_test Was broken, now it's not. Differential Revision: [D68930165](https://our.internmc.facebook.com/intern/diff/D68930165/) ghstack-source-id: 263965167 Pull Request resolved: https://github.com/pytorch/executorch/pull/8084 --- kernels/test/op_upsample_nearest2d_test.cpp | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/kernels/test/op_upsample_nearest2d_test.cpp b/kernels/test/op_upsample_nearest2d_test.cpp index 19ad35455ac..ce6d2ad4d66 100644 --- a/kernels/test/op_upsample_nearest2d_test.cpp +++ b/kernels/test/op_upsample_nearest2d_test.cpp @@ -45,6 +45,12 @@ class OpUpsampleNearest2dTest : public OperatorTest { void test_upsample_nearest2d_dtype() { TensorFactory tf; + if (torch::executor::testing::SupportedFeatures::get()->is_aten && + (DTYPE == ScalarType::Char || DTYPE == ScalarType::Short || + DTYPE == ScalarType::Int || DTYPE == ScalarType::Long)) { + // not supported. + return; + } const auto input = tf.make({1, 1, 2, 2}, {1, 2, 3, 4}); std::array output_size = {4, 4}; auto out = tf.zeros({1, 1, 4, 4});