Removed unnecessary .to() call since it has no effect before .cuda() (#3783)

lounishamroun · svekars · web-flow · commit f38078162a7c · 2026-03-04T09:21:07.000-08:00
Fixes #3782 ## Description Removed unnecessary `.to()` in the `generate_data` function. cc @williamwen42 @msaroufim @anijain2305 @angelayi Co-authored-by: Svetlana Karslioglu <svekars@meta.com>
diff --git a/intermediate_source/torch_compile_full_example.py b/intermediate_source/torch_compile_full_example.py
@@ -79,7 +79,7 @@ def timed(fn):
 # batch size.
 def generate_data(b):
     return (
-        torch.randn(b, 3, 128, 128).to().cuda(),
+        torch.randn(b, 3, 128, 128).cuda(),
         torch.randint(1000, (b,)).cuda(),
     )
 

Original file line number	Diff line number	Diff line change
`@@ -79,7 +79,7 @@ def timed(fn):`
`79`	`79`	`# batch size.`
`80`	`80`	`def generate_data(b):`
`81`	`81`	`return (`
`82`		`- torch.randn(b, 3, 128, 128).to().cuda(),`
	`82`	`+ torch.randn(b, 3, 128, 128).cuda(),`
`83`	`83`	`torch.randint(1000, (b,)).cuda(),`
`84`	`84`	`)`
`85`	`85`