Fix importing issue for inductor_debug_cpu (#2943)

* Fix importing issue for inductor_debug_cpu --------- Co-authored-by: Svetlana Karslioglu <[email protected]>
pytorch · Jul 22, 2024 · 0a5b58b · 0a5b58b
1 parent 2ec46a4
commit 0a5b58b
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
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 
     "beginner_source/onnx/onnx_registry_tutorial", # reenable after 2941 is fixed.
     "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('''