Repository: pytorch
Issue: CompiledFxGraph.current_callable is not thread-safe
Labels: triaged,module: multithreading

### 🐛 Describe the bug

After a new thread compiled, CompiledFxGraph.current_callable is overwritten for all threads.

To repro, please first patch CompiledFxGraph.__call__ as:
```python
    def __call__(self, inputs: Sequence[Any]) -> Any:
        assert self.current_callable is not None
        print(f"thread id: {threading.get_native_id()}, callable id: {id(self.current_callable)}") # new line for callable id
        try:
            return self.current_callable(inputs)
        finally:
            AutotuneCacheBundler.end_compile()
```

Then, run the following code:
```python
import torch
import threading
import time

def foo(x, y):
    return x + y

foo = torch.compile(foo)
count = 0

def wrapper(x, y, odd):
    global count
    for _ in range(10):
        z = foo(x, y)
        # force 2 threads to alternate
        if count % 2 == odd and count < 5:
            time.sleep(1)
        count += 1

threads = []
for _ in range(2):
    threads.append(threading.Thread(target=wrapper, args=(torch.ones(1, device="cuda"), torch.ones(1, device="cuda"), _)))

for thread in threads:
    thread.start()

for thread in threads:
    thread.join()
```

Expected Output:
```
thread id: 124979, callable id: 139811104076800 # Note that this is changed to 139811058639792 later for thread id 124979
thread id: 124980, callable id: 139811058639792
thread id: 124980, callable id: 139811058639792
thread id: 124979, callable id: 139811058639792
thread id: 124980, callable id: 139811058639792
thread id: 124979, callable id: 139811058639792
thread id: 124980, callable id: 139811058639792
thread id: 124980, callable id: 139811058639792
thread id: 124980, callable id: 139811058639792
thread id: 124980, callable id: 139811058639792
thread id: 124980, callable id: 139811058639792
thread id: 124980, callable id: 139811058639792
thread id: 124980, callable id: 139811058639792
thread id: 124979, callable id: 139811058639792
thread id: 124979, callable id: 139811058639792
thread id: 124979, callable id: 139811058639792
thread id: 124979, callable id: 139811058639792
thread id: 124979, callable id: 139811058639792
thread id: 124979, callable id: 139811058639792
thread id: 124979, callable id: 139811058639792
```

This currently blocks #123177.

### Versions

PyTorch version: 2.6.0a0+git2c830c5
Is debug build: False
CUDA used to build PyTorch: 12.0
ROCM used to build PyTorch: N/A

OS: CentOS Stream 9 (x86_64)
GCC version: (GCC) 11.5.0 20240719 (Red Hat 11.5.0-2)
Clang version: Could not collect
CMake version: version 3.26.4
Libc version: glibc-2.34

Python version: 3.10.15 (main, Oct  3 2024, 07:27:34) [GCC 11.2.0] (64-bit runtime)
Python platform: Linux-5.19.0-0_fbk12_hardened_11583_g0bef9520ca2b-x86_64-with-glibc2.34
Is CUDA available: True
CUDA runtime version: 12.0.140
CUDA_MODULE_LOADING set to: LAZY
GPU models and configuration: 
GPU 0: NVIDIA H100
GPU 1: NVIDIA H100
GPU 2: NVIDIA H100
GPU 3: NVIDIA H100

Nvidia driver version: 525.105.17
cuDNN version: Probably one of the following:
/usr/lib64/libcudnn.so.9.1.1
/usr/lib64/libcudnn_adv.so.9.1.1
/usr/lib64/libcudnn_cnn.so.9.1.1
/usr/lib64/libcudnn_engines_precompiled.so.9.1.1
/usr/lib64/libcudnn_engines_runtime_compiled.so.9.1.1
/usr/lib64/libcudnn_graph.so.9.1.1
/usr/lib64/libcudnn_heuristic.so.9.1.1
/usr/lib64/libcudnn_ops.so.9.1.1
HIP runtime version: N/A
MIOpen runtime version: N/A
Is XNNPACK available: True

CPU:
Architecture:                    x86_64
CPU op-mode(s):                  32-bit, 64-bit
Address sizes:                   52 bits physical, 57 bits virtual
Byte Order:                      Little Endian
CPU(s):                          184
On-line CPU(s) list:             0-183
Vendor ID:                       AuthenticAMD
Model name:                      AMD EPYC 9654 96-Core Processor
CPU family:                      25
Model:                           17
Thread(s) per core:              1
Core(s) per socket:              184
Socket(s):                       1
Stepping:                        1
BogoMIPS:                        4792.80
Flags:                           fpu vme de pse tsc msr pae mce cx8 apic sep mtrr pge mca cmov pat pse36 clflush mmx fxsr sse sse2 ht syscall nx mmxext fxsr_opt pdpe1gb rdtscp lm rep_good nopl cpuid extd_apicid tsc_known_freq pni pclmulqdq ssse3 fma cx16 pcid sse4_1 sse4_2 x2apic movbe popcnt tsc_deadline_timer aes xsave avx f16c rdrand hypervisor lahf_lm cmp_legacy svm cr8_legacy abm sse4a misalignsse 3dnowprefetch osvw perfctr_core invpcid_single ssbd ibrs ibpb stibp vmmcall fsgsbase tsc_adjust bmi1 avx2 smep bmi2 erms invpcid avx512f avx512dq rdseed adx smap avx512ifma clflushopt clwb avx512cd sha_ni avx512bw avx512vl xsaveopt xsavec xgetbv1 xsaves avx512_bf16 clzero xsaveerptr wbnoinvd arat npt lbrv nrip_save tsc_scale vmcb_clean pausefilter pfthreshold v_vmsave_vmload vgif avx512vbmi umip pku ospke avx512_vbmi2 gfni vaes vpclmulqdq avx512_vnni avx512_bitalg avx512_vpopcntdq rdpid fsrm arch_capabilities
Virtualization:                  AMD-V
Hypervisor vendor:               KVM
Virtualization type:             full
L1d cache:                       11.5 MiB (184 instances)
L1i cache:                       11.5 MiB (184 instances)
L2 cache:                        92 MiB (184 instances)
L3 cache:                        2.9 GiB (184 instances)
NUMA node(s):                    1
NUMA node0 CPU(s):               0-183
Vulnerability Itlb multihit:     Not affected
Vulnerability L1tf:              Not affected
Vulnerability Mds:               Not affected
Vulnerability Meltdown:          Not affected
Vulnerability Mmio stale data:   Not affected
Vulnerability Retbleed:          Not affected
Vulnerability Spec store bypass: Vulnerable
Vulnerability Spectre v1:        Vulnerable: __user pointer sanitization and usercopy barriers only; no swapgs barriers
Vulnerability Spectre v2:        Vulnerable, IBPB: disabled, STIBP: disabled
Vulnerability Srbds:             Not affected
Vulnerability Tsx async abort:   Not affected

Versions of relevant libraries:
[pip3] flake8==6.1.0
[pip3] flake8-bugbear==23.3.23
[pip3] flake8-comprehensions==3.15.0
[pip3] flake8-executable==2.1.3
[pip3] flake8-logging-format==0.9.0
[pip3] flake8-pyi==23.3.1
[pip3] flake8-simplify==0.19.3
[pip3] mypy==1.11.2
[pip3] mypy-extensions==1.0.0
[pip3] numpy==1.26.0
[pip3] optree==0.13.0
[pip3] pytorch-triton==3.1.0+cf34004b8a
[pip3] torch==2.6.0a0+git2c830c5
[conda] magma-cuda121             2.6.1                         1    pytorch
[conda] numpy                     1.26.0                   pypi_0    pypi
[conda] optree                    0.13.0                   pypi_0    pypi
[conda] pytorch-triton            3.1.0+cf34004b8a          pypi_0    pypi
[conda] torch                     2.6.0a0+git2c830c5           dev_0    <develop>
[conda] torchfix                  0.4.0                    pypi_0    pypi

cc @ezyang @gchanan @zou3519 @kadeng @msaroufim @albanD @chauhang @penguinwu