validate parallel type

Priya2698 · Priya2698 · commit a4d7b8e83d20 · 2026-02-02T19:04:14.000-08:00
diff --git a/csrc/tensor_view.cpp b/csrc/tensor_view.cpp
@@ -769,6 +769,10 @@ TensorView* TensorView::swizzle(
 }
 
 TensorView* TensorView::swizzle1d(int64_t x, ParallelType pt) {
+  NVF_CHECK(
+      deviceParallelTypes().contains(pt),
+      "Swizzle1D only supports device parallel types, given: ",
+      pt);
   domain()->swizzle1d(x, pt);
   return this;
 }

Original file line number	Diff line number	Diff line change
`@@ -769,6 +769,10 @@ TensorView* TensorView::swizzle(`
`769`	`769`	`}`
`770`	`770`
`771`	`771`	`TensorView* TensorView::swizzle1d(int64_t x, ParallelType pt) {`
	`772`	`+ NVF_CHECK(`
	`773`	`+ deviceParallelTypes().contains(pt),`
	`774`	`+ "Swizzle1D only supports device parallel types, given: ",`
	`775`	`+ pt);`
`772`	`776`	`domain()->swizzle1d(x, pt);`
`773`	`777`	`return this;`
`774`	`778`	`}`