[Issue]: hip_global.cpp:158 : Module not initialized.

[songrui2009]

Problem Description

diff --git a/setup.py b/setup.py
index aa0ade9..41926fa 100644
--- a/setup.py
+++ b/setup.py
@@ -9,6 +9,8 @@ from setuptools import find_packages, setup
IS_ROCM = True
ROCM_HOME = "/opt/rocm"

+offload_arch = os.environ.get("PYTORCH_ROCM_ARCH", "gfx1100").replace(";", ",")
+
version = None
exec(open("nerfacc/version.py", "r").read())

@@ -69,7 +71,7 @@ def get_extensions():
extra_compile_args["cxx"] += ["-arch", "arm64"]
extra_link_args += ["-arch", "arm64"]

•    hipcc_flags = ["-O3", "-D__HIP_PLATFORM_AMD__", "-DUSE_ROCM" , "--offload-arch=gfx942"]
  

•    hipcc_flags = ["-O3", "-D__HIP_PLATFORM_AMD__", "-DUSE_ROCM" , f"--offload-arch={offload_arch}"]
  
     if torch.version.hip:
         # USE_ROCM was added to later versions of PyTorch.
  

Operating System

Ubuntu 22.04.5 LTS

CPU

AMD Ryzen 5 3600 6-Core Processor

GPU

Radeon RX 7900 XT

ROCm Version

rocm-6.4.3

ROCm Component

No response

Steps to Reproduce

No response

(Optional for Linux users) Output of /opt/rocm/bin/rocminfo --support

No response

Additional Information

No response