From e430d55ae6878b0f7d27fbcc1ad47d67779aaef7 Mon Sep 17 00:00:00 2001 From: GavinWongYF <61688522+GavinWongYF@users.noreply.github.com> Date: Sat, 4 Apr 2026 00:42:05 -0400 Subject: [PATCH] Updated deprecated use_gpu profiler argument to use_device --- beginner_source/introyt/autogradyt_tutorial.py | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) diff --git a/beginner_source/introyt/autogradyt_tutorial.py b/beginner_source/introyt/autogradyt_tutorial.py index abf75a7d266..c065aaebd56 100644 --- a/beginner_source/introyt/autogradyt_tutorial.py +++ b/beginner_source/introyt/autogradyt_tutorial.py @@ -473,16 +473,14 @@ def add_tensors2(x, y): # device = torch.device('cpu') -run_on_gpu = False if torch.cuda.is_available(): device = torch.device('cuda') - run_on_gpu = True - + x = torch.randn(2, 3, requires_grad=True) y = torch.rand(2, 3, requires_grad=True) z = torch.ones(2, 3, requires_grad=True) -with torch.autograd.profiler.profile(use_cuda=run_on_gpu) as prf: +with torch.autograd.profiler.profile(use_device=device.type) as prf: for _ in range(1000): z = (z / x) * y