Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 2 additions & 0 deletions python/test/unit/language/test_core.py
Original file line number Diff line number Diff line change
Expand Up @@ -2793,6 +2793,8 @@ def test_dot(M, N, K, num_warps, col_a, col_b, epilogue, allow_tf32, in_dtype, o
pytest.skip("Only test out_dtype=float16 on devices with sm >=80")
if capability[0] < 9 and in_dtype == 'float8e4nv':
pytest.skip("float8e4nv not supported on sm <= 80")
if is_hip() and (in_dtype == 'float8e4nv' or in_dtype == 'float8e5'):
pytest.skip("float8e4nv and float8e5 not supported on HIP")
if is_interpreter() and in_dtype == 'int8':
pytest.skip(
"numpy.dot with int8 inputs will overflow while tl.dot doesn't because MMA instruction's accumulator is 32-bit"
Expand Down
6 changes: 3 additions & 3 deletions third_party/amd/backend/compiler.py
Original file line number Diff line number Diff line change
Expand Up @@ -82,12 +82,12 @@ def hash(self):
class HIPBackend(BaseBackend):

@staticmethod
def supports_target(target: tuple):
def supports_target(target: list):
return target[0] == 'hip'

def __init__(self, target: tuple) -> None:
def __init__(self, target: list) -> None:
super().__init__(target)
assert isinstance(target, tuple) and len(target) == 3
assert isinstance(target, list) and len(target) == 3
assert isinstance(target[1], str)
self.binary_ext = "hsaco"

Expand Down
2 changes: 1 addition & 1 deletion third_party/amd/backend/driver.py
Original file line number Diff line number Diff line change
Expand Up @@ -272,4 +272,4 @@ def get_current_target(self):
device_properties = self.utils.get_device_properties(device)
arch = device_properties['arch']
warpSize = device_properties['warpSize']
return ("hip", arch.split(':')[0], warpSize)
return ["hip", arch.split(':')[0], warpSize]