From c477b99d2189f70dff5b6c72455866264ca2eaaf Mon Sep 17 00:00:00 2001
From: Scott Wolchok <swolchok@meta.com>
Date: Thu, 30 Jan 2025 16:38:55 -0800
Subject: [PATCH] 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-poisoned]
---
 kernels/test/op_upsample_bilinear2d_test.cpp | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/kernels/test/op_upsample_bilinear2d_test.cpp b/kernels/test/op_upsample_bilinear2d_test.cpp
index f17b83845ec..02307d3d2f0 100644
--- a/kernels/test/op_upsample_bilinear2d_test.cpp
+++ b/kernels/test/op_upsample_bilinear2d_test.cpp
@@ -44,6 +44,12 @@ class OpUpsampleBilinear2dTest : public OperatorTest {
   void test_upsample_bilinear2d_dtype() {
     TensorFactory<DTYPE> 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, 1, 2}, {1, 4});
     std::array<int64_t, 2> output_size = {1, 4};
     auto out = tf.zeros({1, 1, 1, 4});