diff --git a/kernels/test/op_unbind_copy_test.cpp b/kernels/test/op_unbind_copy_test.cpp index 3e861e29a71..4901b0f69e3 100644 --- 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 tf; TensorListFactory tlf;