Skip unbind_copy ATen mode failure test

izaitsevfb · web-flow · commit 307c7b2a9f6e · 2025-01-22T12:09:07.000-08:00
Differential Revision: D68456720 Pull Request resolved: #7805
diff --git a/kernels/test/op_unbind_copy_test.cpp b/kernels/test/op_unbind_copy_test.cpp
@@ -269,6 +269,9 @@ TEST_F(OpUnbindCopyIntOutTest, UnbindWorksWithZeroSizedTensors) {
 }
 
 TEST_F(OpUnbindCopyIntOutTest, UnbindFailsWithWronglyAllocatedOutput) {
+  if (torch::executor::testing::SupportedFeatures::get()->is_aten) {
+    GTEST_SKIP() << "ATen kernel can handle mismatched output shape";
+  }
   TensorFactory<ScalarType::Int> tf;
   TensorListFactory<ScalarType::Int> tlf;
 

Original file line number	Diff line number	Diff line change
`@@ -269,6 +269,9 @@ TEST_F(OpUnbindCopyIntOutTest, UnbindWorksWithZeroSizedTensors) {`
`269`	`269`	`}`
`270`	`270`
`271`	`271`	`TEST_F(OpUnbindCopyIntOutTest, UnbindFailsWithWronglyAllocatedOutput) {`
	`272`	`+ if (torch::executor::testing::SupportedFeatures::get()->is_aten) {`
	`273`	`+ GTEST_SKIP() << "ATen kernel can handle mismatched output shape";`
	`274`	`+ }`
`272`	`275`	`TensorFactory<ScalarType::Int> tf;`
`273`	`276`	`TensorListFactory<ScalarType::Int> tlf;`
`274`	`277`