viable/strict/1789252151: [inductor] Gate CUDA TMA tests on CUDA devices (#196636)
- PyTorch 近 90 天出现 850 次
- PyTorch 近 90 天第 839 次发版
- 上一次:同一天稍早 · viable/strict/1789237655: Upgrade Nvidia's Windows CRCR Repo to L2 (#196602)
发生了什么
has_triton_tma_device() reports tensor-descriptor support when any installed Triton backend supports it, including CPU and XPU. A ROCm build that also bundles the Triton CPU backend can therefore return true even though its AMD GPU cannot run CUDA TMA templates. Add a CUDA-specific predicate requiring a non-HIP CUDA device with compute capability 9.0 or newer, and use it for the CUDA-only TMA tests. Keep the existin…
摘要按规则整理自下方来源原文