Merge branch 'main' into titaiwang/adapt_optimizer_changes_to_graphs

svekars · web-flow · commit cb73ad98519d · 2024-07-22T11:27:55.000-07:00
diff --git a/.jenkins/validate_tutorials_built.py b/.jenkins/validate_tutorials_built.py
@@ -54,7 +54,6 @@
     "intermediate_source/flask_rest_api_tutorial",
     "intermediate_source/text_to_speech_with_torchaudio",
     "intermediate_source/tensorboard_profiler_tutorial", # reenable after 2.0 release.
-    "intermediate_source/inductor_debug_cpu", # reenable after 2942 
     "intermediate_source/torch_export_tutorial" # reenable after 2940 is fixed.
 ]
 
diff --git a/intermediate_source/inductor_debug_cpu.py b/intermediate_source/inductor_debug_cpu.py
@@ -110,7 +110,8 @@ def forward1(self, arg0_1, arg1_1):
 # C++ kernel in ``output_code``:
 #
 
-from torch._inductor.codecache import AsyncCompile
+import torch
+from torch._inductor.async_compile import AsyncCompile
 async_compile = AsyncCompile()
 
 cpp_fused_cat_maximum_neg_0 = async_compile.cpp('''

Original file line number	Diff line number	Diff line change
`@@ -54,7 +54,6 @@`
`54`	`54`	`"intermediate_source/flask_rest_api_tutorial",`
`55`	`55`	`"intermediate_source/text_to_speech_with_torchaudio",`
`56`	`56`	`"intermediate_source/tensorboard_profiler_tutorial", # reenable after 2.0 release.`
`57`		`- "intermediate_source/inductor_debug_cpu", # reenable after 2942`
`58`	`57`	`"intermediate_source/torch_export_tutorial" # reenable after 2940 is fixed.`
`59`	`58`	`]`
`60`	`59`
Original file line number	Diff line number	Diff line change
`@@ -110,7 +110,8 @@ def forward1(self, arg0_1, arg1_1):`
`110`	`110`	# C++ kernel in ``output_code``:
`111`	`111`	`#`
`112`	`112`
`113`		`-from torch._inductor.codecache import AsyncCompile`
	`113`	`+import torch`
	`114`	`+from torch._inductor.async_compile import AsyncCompile`
`114`	`115`	`async_compile = AsyncCompile()`
`115`	`116`
`116`	`117`	`cpp_fused_cat_maximum_neg_0 = async_compile.cpp('''`