specify cuda

whoisj · whoisj · commit a40b556a85bc · 2025-12-19T19:43:06.000-05:00
diff --git a/qa/common/gen_qa_models.py b/qa/common/gen_qa_models.py
@@ -1412,8 +1412,8 @@ def forward(self, INPUT0, INPUT1):
     ep = torch.export.export(
         AddSubNet(swap),
         (
-            torch.randn(*input_shape),
-            torch.randn(*input_shape),
+            torch.randn(*input_shape, device="cuda"),
+            torch.randn(*input_shape, device="cuda"),
         ),
     )
     torch._inductor.aoti_compile_and_package(

Original file line number	Diff line number	Diff line change
`@@ -1412,8 +1412,8 @@ def forward(self, INPUT0, INPUT1):`
`1412`	`1412`	`ep = torch.export.export(`
`1413`	`1413`	`AddSubNet(swap),`
`1414`	`1414`	`(`
`1415`		`- torch.randn(*input_shape),`
`1416`		`- torch.randn(*input_shape),`
	`1415`	`+ torch.randn(*input_shape, device="cuda"),`
	`1416`	`+ torch.randn(*input_shape, device="cuda"),`
`1417`	`1417`	`),`
`1418`	`1418`	`)`
`1419`	`1419`	`torch._inductor.aoti_compile_and_package(`