Fix imports

malfet · malfet · commit 41682cc47ca4 · 2024-08-08T16:59:43.000+01:00
diff --git a/test/smoke_test/smoke_test.py b/test/smoke_test/smoke_test.py
@@ -260,7 +260,7 @@ def foo(x: torch.Tensor) -> torch.Tensor:
 
     # Check that SIMD were detected for the architecture
     if device == "cpu":
-        from torch._inductor.cpu_vec_isa import invalid_vec_isa, pick_vec_isa
+        from torch._inductor.codecache import pick_vec_isa, invalid_vec_isa
         isa = pick_vec_isa()
         if isa == invalid_vec_isa:
             raise RuntimeError("Can't detect vectorized ISA for CPU")
