github/ROCm
1
1
Fork 0
mirror of https://github.com/ROCm/ROCm.git synced 2026-04-05 03:01:17 -04:00
Code Issues Packages Projects Releases Wiki Activity

Update hip_backend to use libhsa-runtime for arch info, (#411)

Browse Source 
brings in path changes for pytorch triton wheels

Co-authored-by: jayfurmanek <Jason.Furmanek@amd.com>
This commit is contained in:
Jack Taylor
2023-12-21 15:40:57 +00:00
committed by GitHub
parent 0248bdb29d
commit 1e2fd0dd1a
8 changed files with 21 additions and 23 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
python/triton/common/build.py
View File

@@ -36,7 +36,7 @@ def libcuda_dirs():
@functools.lru_cache()
def rocm_path_dir():
default_path = os.path.join(os.path.dirname(__file__), "..", "third_party", "rocm")
default_path = os.path.join(os.path.dirname(__file__), "..", "third_party", "hip")
# Check if include files have been populated locally. If so, then we are
# most likely in a whl installation and he rest of our libraries should be here
if (os.path.exists(default_path+"/include/hip/hip_runtime.h")):

1
python/triton/compiler/compiler.py
View File

@@ -386,7 +386,6 @@ def get_arch_default_num_stages(device_type, capability=None):
assert _device_backend
arch = _device_backend.get_architecture_descriptor()
num_stages = arch["num_stages"]
return num_stages

12
python/triton/third_party/hip/hip_backend.py vendored
View File

@@ -300,16 +300,14 @@ def gpu_matrix_core_version() -> int:
def get_amdgpu_arch_fulldetails():
# print("get_amdgpu_arch_fulldetails")
"""
get the amdgpu fulll ISA details for compiling:
get the amdgpu full ISA details for compiling:
i.e., arch_triple: amdgcn-amd-amdhsa; arch_name: gfx906; arch_features: sramecc+:xnack-
"""
try:
# TODO: package rocm.cc with Triton
rocm_path_dir = os.getenv("ROCM_PATH", default="/opt/rocm")
rocminfo = subprocess.check_output(rocm_path_dir + '/bin/rocminfo').decode()
gfx_arch_details = re.search('amd.*', rocminfo).group(0).strip().split('--')
arch_info = _triton.get_arch_info()
gfx_arch_details = re.search('amd.*', arch_info).group(0).strip().split('--')
arch_triple = gfx_arch_details[0]
arch_name_features = gfx_arch_details[1].split(':')
arch_name = arch_name_features[0]
@@ -319,13 +317,13 @@ def get_amdgpu_arch_fulldetails():
gfx_arch = os.environ.get('MI_GPU_ARCH', arch_name)
if gfx_arch is None:
raise RuntimeError('gfx_arch is None (not specified)')
mat_core_ver = gpu_matrix_core_version()
capability = gpu_matrix_core_version() * 100
return {"gfx_triple": arch_triple, "gfx_arch": gfx_arch, "gfx_features": arch_features,\
"capability": capability, "matrix_core_version": mat_core_ver}
except BaseException:
except BaseException as e:
print("Error: Attempting to get amgpu ISA Details {}".format(e))
return None

4
python/triton/third_party/hip/lib/Target/HSACO/HSACOTranslation.cpp vendored
View File

@@ -217,7 +217,7 @@ std::string generate_hsaco(llvm::Module *module, const std::string &triple,
.parent_path()
.parent_path() /
"triton" / "third_party" /
"rocm" / "llvm" / "bin" / "ld.lld";
"hip" / "llvm" / "bin" / "ld.lld";
std::string lld_path = compiletime_path.string();
if (!std::filesystem::exists(lld_path)) {
std::string rocm_path = ::triton::tools::getenv("ROCM_PATH");
@@ -778,4 +778,4 @@ translateTritonIRToHSACO(mlir::ModuleOp module, std::string gfx_arch,
}
} // namespace triton
} // namespace mlir
} // namespace mlir