diff --git a/openequivariance/CMakeLists.txt b/openequivariance/CMakeLists.txt
index 232233f..e7fac87 100644
--- a/openequivariance/CMakeLists.txt
+++ b/openequivariance/CMakeLists.txt
@@ -158,8 +158,8 @@ if(hip_FOUND)
     )
 
     set(HIP_LINK_LIBS
-        hiprtc
         hip_stub_lib
+	    hip::host
     )
     add_stable_extension(torch_stable_hip HIP_BACKEND "${HIP_LINK_LIBS}")
 endif()
diff --git a/openequivariance/openequivariance/_torch/extlib/__init__.py b/openequivariance/openequivariance/_torch/extlib/__init__.py
index ef17724..7326230 100644
--- a/openequivariance/openequivariance/_torch/extlib/__init__.py
+++ b/openequivariance/openequivariance/_torch/extlib/__init__.py
@@ -89,7 +89,6 @@ def load_jit_extension():
 
             extra_cflags.append("-DCUDA_BACKEND")
         elif torch.version.hip:
-            extra_link_args.extend(["-lhiprtc"])
             torch_libs = library_paths("cuda")[0]
             extra_link_args.append("-Wl,-rpath," + torch_libs)
             extra_cflags.append("-DHIP_BACKEND")
diff --git a/openequivariance/openequivariance/extension/libtorch_tp_jit_stable.cpp b/openequivariance/openequivariance/extension/libtorch_tp_jit_stable.cpp
index 440b764..abece92 100644
--- a/openequivariance/openequivariance/extension/libtorch_tp_jit_stable.cpp
+++ b/openequivariance/openequivariance/extension/libtorch_tp_jit_stable.cpp
@@ -11,9 +11,6 @@
 #include <torch/headeronly/util/shim_utils.h>
 #include <torch/csrc/inductor/aoti_torch/c/shim.h>
 
-#ifdef HIP_BACKEND
-    #include <c10/hip/HIPStream.h>
-#endif
 
 using Tensor = torch::stable::Tensor;
 using Dtype = torch::headeronly::ScalarType;