From 372d587800cbf93ef1bcf064bbda34c5d2d1f6a4 Mon Sep 17 00:00:00 2001 From: Lounis Date: Fri, 27 Feb 2026 23:27:10 +0100 Subject: [PATCH] Remove redundant .to() call in generate_data() --- intermediate_source/torch_compile_full_example.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/intermediate_source/torch_compile_full_example.py b/intermediate_source/torch_compile_full_example.py index 7804d5b155..84e8f74e76 100644 --- 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(), )