Fix installation for amd gpu

install.py

@@ -8,7 +8,9 @@
 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__)

tritonbench/operators/bf16xint16_gemm/kernel.py

@@ -6,6 +6,7 @@
 
 import triton
 import triton.language as tl
+
 from tritonbench.utils.env_utils import is_cuda
 
 

tritonbench/operators/fp8_gemm/persistent.py

@@ -4,6 +4,7 @@
 import triton
 import triton.language as tl
 import triton.tools.experimental_descriptor
+
 from tritonbench.utils.env_utils import is_cuda
 
 cublas = None

tritonbench/operators/fp8_gemm/tutorial.py

@@ -153,6 +153,7 @@
 
 import triton
 import triton.language as tl
+
 from tritonbench.utils.env_utils import is_cuda
 
 

tritonbench/operators/gemm/persistent_matmul.py

@@ -3,6 +3,7 @@
 import torch
 import triton
 import triton.language as tl
+
 from tritonbench.utils.env_utils import is_cuda
 from tritonbench.utils.triton_op import IS_FBCODE