Fix installation for amd gpu (#128)

Summary:
Fix #127

The is_hip function has been moved to tritonbench.utils.env_utils.

Pull Request resolved: #128

Reviewed By: xuzhao9

Differential Revision: D68031969

Pulled By: FindHao

fbshipit-source-id: 229c6ab20953b893f2f1fc7da7663e6feddde53a

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