AI/rocm-systems
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

libhsakmt: Implement HSA_FORCE_ASIC_TYPE to overwrite asic type

Browse Source 
Force all the GPUs to a certain type, use the below command:
HSA_FORCE_ASIC_TYPE="10.1.0 1 gfx1010 14"
meaning major.minor.step dgpu asic_name asic_id

This will faciliate the cooperation across the teams for bringing up
ASICs which reuse existing device IDs.

Change-Id: I40fe4c9b46d3ccb3e38ea52250e80e82fb50fb0f
Signed-off-by: Yong Zhao <Yong.Zhao@amd.com>


[ROCm/ROCR-Runtime commit: d6539ddc24]
This commit is contained in:
Yong Zhao
2019-09-04 19:03:11 -04:00
committed by Yong Zhao
parent cc050951de
commit fab21d6d81
3 changed files with 71 additions and 25 deletions
Download Patch File Download Diff File Expand all files Collapse all files
projects/rocr-runtime/src/openclose.c
+32 -2
View File
@@ -82,7 +82,7 @@ static inline void init_page_size(void)
PAGE_SHIFT = ffs(PAGE_SIZE) - 1;
}
static void init_vars_from_env(void)
static HSAKMT_STATUS init_vars_from_env(void)
{
char *envvar;
int debug_level;
@@ -104,6 +104,34 @@ static void init_vars_from_env(void)
envvar = getenv("HSA_ZFB");
if (envvar)
zfb_support = atoi(envvar);
/* Force all the GPUs to a certain type, use the below command:
* export HSA_FORCE_ASIC_TYPE="10.1.0 1 Navi10 14"
* meaning major.minor.step dgpu asic_name asic_id
*/
envvar = getenv("HSA_FORCE_ASIC_TYPE");
if (envvar) {
uint32_t major, minor, step, dgpu, asic_family;
if ((sscanf(envvar, "%u.%u.%u %u %63s %u", &major, &minor, &step,
&dgpu, force_asic_name, &asic_family) != 6)
|| (major > 63 || minor > 255 || step > 255)
|| dgpu > 1 || asic_family >= CHIP_LAST) {
pr_err("HSA_FORCE_ASIC_TYPE %s is invalid\n", envvar);
return HSAKMT_STATUS_ERROR;
}
force_asic_entry.major = major;
force_asic_entry.minor = minor;
force_asic_entry.stepping = step;
force_asic_entry.is_dgpu = dgpu;
force_asic_entry.asic_family = asic_family;
force_asic = 1;
}
return HSAKMT_STATUS_SUCCESS;
}
HSAKMT_STATUS HSAKMTAPI hsaKmtOpenKFD(void)
@@ -122,7 +150,9 @@ HSAKMT_STATUS HSAKMTAPI hsaKmtOpenKFD(void)
clear_after_fork();
if (kfd_open_count == 0) {
init_vars_from_env();
result = init_vars_from_env();
if (result != HSAKMT_STATUS_SUCCESS)
goto open_failed;
fd = open(kfd_device_name, O_RDWR | O_CLOEXEC);