def test_igemm_hpa_hip_tn(tmpdir):
Tensile.Tensile([
Tensile.TensileTestPath("pre_checkin/igemm_hpa_hip_tn.yaml"),
tmpdir.strpath
])
def test_sgemm_gsu(tmpdir):
Tensile.Tensile([Tensile.TensileTestPath("nightly/global_split_u/sgemm_gsu.yaml"), tmpdir.strpath])
def test_dgemm_asm(tmpdir):
Tensile.Tensile([
Tensile.TensileTestPath("pre_checkin/dgemm_asm.yaml"), tmpdir.strpath
])
def test_sgemm_asm_nt(tmpdir):
Tensile.Tensile([
Tensile.TensileTestPath("nightly/nonbatched/sgemm_asm_nt.yaml"),
tmpdir.strpath
])
def test_create_library(tmpdir):
Tensile.Tensile([
Tensile.TensileTestPath("disabled/test_create_library.yaml"),
tmpdir.strpath
])
def test_hgemm_asm_tn(tmpdir):
Tensile.Tensile([
Tensile.TensileTestPath("pre_checkin/hgemm_asm_tn.yaml"),
tmpdir.strpath
])
def test_cgemm_hip_source_cc(tmpdir):
Tensile.Tensile([
Tensile.TensileTestPath(
"pre_checkin/float_complex/cgemm_hip_source_cc.yaml"),
tmpdir.strpath
])
def test_persistent(tmpdir):
Tensile.Tensile([
Tensile.TensileTestPath("nightly/classic/test_persistent.yaml"),
tmpdir.strpath
])
def test_convolution(tmpdir):
Tensile.Tensile([
Tensile.TensileTestPath("nightly/classic/test_convolution.yaml"),
tmpdir.strpath
])
def test_dgemm_defaults(tmpdir):
Tensile.Tensile([
Tensile.TensileTestPath("pre_checkin/source/test_dgemm_defaults.yaml"),
tmpdir.strpath
])
def test_config(tensile_args, config, tmpdir):
Tensile.Tensile([config, tmpdir.strpath, *tensile_args])
def test_stridea0_pack_nt(tmpdir):
Tensile.Tensile([
Tensile.TensileTestPath("disabled/stridea0_pack_nt.yaml"),
tmpdir.strpath
])
def test_assertion_selection(tmpdir):
Tensile.Tensile([
Tensile.TensileTestPath("disabled/test_assertion_selection.yaml"),
tmpdir.strpath
])
def test_starter_packed_case(tmpdir):
Tensile.Tensile([
Tensile.TensileTestPath("disabled/starter_packed_case.yaml"),
tmpdir.strpath
])
def test_hgemm_hpa_iu2_asm_nt(tmpdir):
Tensile.Tensile([
Tensile.TensileTestPath("pre_checkin/hgemm_hpa_iu2_asm_nt.yaml"),
tmpdir.strpath
])
def test_tensor_contraction(tmpdir):
Tensile.Tensile([
Tensile.TensileTestPath(
"nightly/classic/test_tensor_contraction.yaml"), tmpdir.strpath
])
def test_sgemm_asm_tn_bigk(tmpdir):
Tensile.Tensile([
Tensile.TensileTestPath("pre_checkin/sgemm_asm_tn_bigk.yaml"),
tmpdir.strpath
])
def test_bfloat16_source_tn(tmpdir):
Tensile.Tensile([Tensile.TensileTestPath("pre_checkin/bfloat16/bfloat16_hpa_source_tn.yaml"), tmpdir.strpath])
def test_cgemm_asm(tmpdir):
Tensile.Tensile([
Tensile.TensileTestPath("pre_checkin/float_complex/cgemm_asm.yaml"),
tmpdir.strpath
])
def test_dgemm_lsu(tmpdir):
Tensile.Tensile([
Tensile.TensileTestPath("nightly/local_split_u/dgemm_lsu.yaml"),
tmpdir.strpath
])
def test_hgemm_hpa_dot2_tn_2(tmpdir):
Tensile.Tensile([
Tensile.TensileTestPath("nightly/dot2/hgemm_hpa_dot2_tn_2.yaml"),
tmpdir.strpath
])
def test_hgemm_lsu_grvw1(tmpdir):
Tensile.Tensile([
Tensile.TensileTestPath("nightly/local_split_u/hgemm_lsu_grvw1.yaml"),
tmpdir.strpath
])
def run(conv, problemType, dataType='s'):
configFile = run_generate_yaml(conv, problemType, dataType)
Tensile.Tensile([str(configFile), str(tmp_path), *tensile_args])
def test_hgemm_nn_source(tmpdir):
Tensile.Tensile([
Tensile.TensileTestPath("disabled/hgemm_nn_source.yaml"),
tmpdir.strpath
])
