Dynamo 中 Triton CUDA 编译失败 | AI生成和翻译
文件 “/home/lzw/.local/lib/python3.11/site-packages/torch/utils/_triton.py”，第103行，在 triton_backend 中 target = driver.active.get_current_target() ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ 文件 “/home/lzw/.local/lib/python3.11/site-packages/triton/runtime/driver.py”，第23行，在 getattr 中 self._initialize_obj() 文件 “/home/lzw/.local/lib/python3.11/site-packages/triton/runtime/driver.py”，第20行，在 _initialize_obj 中 self._obj = self._init_fn() ^^^^^^^^^^^^^^^ 文件 “/home/lzw/.local/lib/python3.11/site-packages/triton/runtime/driver.py”，第9行，在_create_driver 中 return actives0 ^^^^^^^^^^^^ 文件 “/home/lzw/.local/lib/python3.11/site-packages/triton/backends/nvidia/driver.py”，第450行，在 init 中 self.utils = CudaUtils() # TODO: 改为静态 ^^^^^^^^^^^ 文件 “/home/lzw/.local/lib/python3.11/site-packages/triton/backends/nvidia/driver.py”，第80行，在 init 中 mod = compile_module_from_src(Path(os.path.join(dirname, “driver.c”)).read_text(), “cuda_utils”) ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ 文件 “/home/lzw/.local/lib/python3.11/site-packages/triton/backends/nvidia/driver.py”，第57行，在 compile_module_from_src 中 so =_build(name, src_path, tmpdir, library_dirs(), include_dir, libraries) ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ 文件 “/home/lzw/.local/lib/python3.11/site-packages/triton/runtime/build.py”，第50行，在_build 中 ret = subprocess.check_call(cc_cmd) ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ 文件 “/usr/lib/python3.11/subprocess.py”，第413行，在 check_call 中 raise CalledProcessError(retcode, cmd) torch._dynamo.exc.BackendCompilerFailed: backend=’inductor’ 引发: CalledProcessError: 命令 ‘[‘/usr/bin/gcc’, ‘/tmp/tmpt2wndipb/main.c’, ‘-O3’, ‘-shared’, ‘-fPIC’, ‘-Wno-psabi’, ‘-o’, ‘/tmp/tmpt2wndipb/cuda_utils.cpython-311-x86_64-linux-gnu.so’, ‘-lcuda’, ‘-L/home/lzw/.local/lib/python3.11/site-packages/triton/backends/nvidia/lib’, ‘-L/lib/x86_64-linux-gnu’, ‘-L/lib/i386-linux-gnu’, ‘-I/home/lzw/.local/lib/python3.11/site-packages/triton/backends/nvidia/include’, ‘-I/tmp/tmpt2wndipb’, ‘-I/usr/include/python3.11’]’ 返回非零退出状态 1。
设置 TORCH_LOGS=”+dynamo” 和 TORCHDYNAMO_VERBOSE=1 以获取更多信息
你可以通过以下设置抑制此异常并回退到 eager 模式： import torch._dynamo torch._dynamo.config.suppress_errors = True