43 lines
1.3 KiB
Python
43 lines
1.3 KiB
Python
|
import os, random
|
||
|
|
|
||
|
|
class TestSpec:
|
||
|
|
def prepare(self, device, seed):
|
||
|
|
raise NotImplementedError("prepare must be implemented in the derived class")
|
||
|
|
def get_exec_state(self):
|
||
|
|
raise NotImplementedError("get_exec_state must be implemented in the derived class")
|
||
|
|
def name(self): return self.__class__.__name__
|
||
|
|
|
||
|
|
class DeviceSpec:
|
||
|
|
def prepare(self, seed):
|
||
|
|
raise NotImplementedError("prepare must be implemented in the derived class")
|
||
|
|
def get_exec_state(self):
|
||
|
|
raise NotImplementedError("get_exec_state must be implemented in the derived class")
|
||
|
|
def name(self): return self.__class__.__name__
|
||
|
|
|
||
|
|
class HCQSpec(DeviceSpec): pass
|
||
|
|
class AMDSpec(HCQSpec):
|
||
|
|
def __init__(self):
|
||
|
|
assert os.path.exists('/sys/module/amdgpu'), "amdgpu module should be loaded"
|
||
|
|
|
||
|
|
def prepare(self, seed):
|
||
|
|
self.env = {
|
||
|
|
"AMD": 1,
|
||
|
|
"AMD_LLVM": 0
|
||
|
|
}
|
||
|
|
|
||
|
|
def get_exec_state(self): return self.env
|
||
|
|
|
||
|
|
class AMSpec(AMDSpec):
|
||
|
|
def __init__(self):
|
||
|
|
assert not os.path.exists('/sys/module/amdgpu'), "amdgpu module should not be loaded"
|
||
|
|
|
||
|
|
def prepare(self, seed):
|
||
|
|
super().prepare(seed)
|
||
|
|
|
||
|
|
self.env = {
|
||
|
|
**self.env, # from AMDSpec
|
||
|
|
"AMD_SDMA_BIND": random.randint(0, 1),
|
||
|
|
"AMD_ALLOC_QUEUE_DEV_MEM": 0, # random.randint(0, 1) need to validate
|
||
|
|
"AMD_QUEUE_SIZE": 1 << random.randint(10, 26),
|
||
|
|
}
|