[Torch 2.11] Guard torch._C._cpu attribute checks for forward compatibility (#35673)

Signed-off-by: atalman <atalman@fb.com>
2026-03-17 14:47:59 -04:00
parent c5030c439d
commit 68f783a727
9 changed files with 10 additions and 12 deletions
--- a/benchmarks/kernels/cpu/benchmark_cpu_attn.py
+++ b/benchmarks/kernels/cpu/benchmark_cpu_attn.py
@@ -27,7 +27,7 @@ def get_attn_isa(
    else:
        if current_platform.get_cpu_architecture() == CpuArchEnum.ARM:
            return "neon"
-        elif torch._C._cpu._is_amx_tile_supported():
+        elif torch.cpu._is_amx_tile_supported():
            return "amx"
        else:
            return "vec"