fix #127

pytorch-labs · Jan 10, 2025 · cab2e94 · cab2e94
1 parent 9b7cca2
commit cab2e94
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/install.py b/install.py
@@ -8,7 +8,8 @@
 from tools.cuda_utils import CUDA_VERSION_MAP, DEFAULT_CUDA_VERSION
 from tools.git_utils import checkout_submodules
 from tools.python_utils import pip_install_requirements
-from tools.torch_utils import is_hip
+from tritonbench.utils.env_utils import is_hip
+
 
 logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)