Update on "NT support for cat with dim > 0 when representable as jagged"

jbschlosser · jbschlosser · commit 52a7fa3ceb7b · 2023-09-01T13:11:35.000-04:00
Used in SAM

[ghstack-poisoned]
diff --git a/test/test_nestedtensor.py b/test/test_nestedtensor.py
@@ -2101,6 +2101,11 @@ def test_narrow(self, device, dtype):
             with self.assertRaisesRegex(RuntimeError, "only dim=0 supported for nested tensors"):
                 nt.narrow(dim=dim, start=0, length=1)
 
+        # error case: non-contiguous NT
+        _, nt_noncont = random_nt_noncontiguous_pair((2, 3, 4))
+        with self.assertRaisesRegex(RuntimeError, "only contiguous nested tensors supported"):
+            nt_noncont.narrow(dim=0, start=0, length=1)
+
     @parametrize("input_dim", [3, 4])
     def test_scaled_dot_product_attention(self, device, input_dim):
 
