ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-05-14 18:17:44 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merge commit 'df54667102a3a1183fa55872eb6889717b42fde6' into develop

Browse Source
This commit is contained in:
assistant-librarian[bot]
2025-06-17 22:13:19 +00:00
parent 0b1e5e1da4
commit c8aa6b376e
11 changed files with 37 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
example/ck_tile/18_flatmm/script/smoke_test_basic.sh
View File

@@ -1,4 +1,8 @@
#!/bin/bash
# Copyright © Advanced Micro Devices, Inc., or its affiliates.
# SPDX-License-Identifier: MIT
EXE="$(find . -name tile_example_flatmm_basic -type f | head -n 1)"
KNAME=1

5
example/ck_tile/35_batched_transpose/script/perf_test.sh
View File

@@ -1,5 +1,8 @@
#!/bin/sh
# Copyright © Advanced Micro Devices, Inc., or its affiliates.
# SPDX-License-Identifier: MIT
EXE=./build/bin/tile_example_batched_transpose
for pr in "fp8" "fp16" "bf16"; do
@@ -8,4 +11,4 @@ $EXE -pr=$pr -N=1 -C=1024 -H=1 -W=1024 -layout_in='NCHW' -layout_out='NHWC'
$EXE -pr=$pr -N=1 -C=1024 -H=1 -W=2048 -layout_in='NCHW' -layout_out='NHWC'
$EXE -pr=$pr -N=1 -C=4096 -H=1 -W=2048 -layout_in='NCHW' -layout_out='NHWC'
done
done

4
example/ck_tile/35_batched_transpose/script/run_full_test.sh
View File

@@ -1,4 +1,8 @@
#!/bin/bash
# Copyright © Advanced Micro Devices, Inc., or its affiliates.
# SPDX-License-Identifier: MIT
#
# in order to run this script you'd first need to build the tile_example_batched_transpose executables in ../build/bin/
#

5
example/ck_tile/35_batched_transpose/script/smoke_test.sh
View File

@@ -1,5 +1,8 @@
#!/bin/sh
# Copyright © Advanced Micro Devices, Inc., or its affiliates.
# SPDX-License-Identifier: MIT
EXE=./build/bin/tile_example_batched_transpose
for pr in "fp8" "fp16" "bf16"; do
@@ -24,4 +27,4 @@ $EXE -pr=$pr -N=8 -C=16 -H=8 -W=16 -layout_in='NHWC' -layout_out='NCHW'
$EXE -pr=$pr -N=1 -C=64 -H=1 -W=1024 -layout_in='NCHW' -layout_out='NHWC'
$EXE -pr=$pr -N=1 -C=64 -H=1024 -W=1 -layout_in='NHWC' -layout_out='NCHW'
done
done

3
test/batched_gemm_softmax_gemm_permute/test_batched_gemm_device_utils.hpp
View File

@@ -1,5 +1,8 @@
#pragma once
// Copyright © Advanced Micro Devices, Inc., or its affiliates.
// SPDX-License-Identifier: MIT
#include <hip/hip_runtime.h>
#include <string>

3
test/gemm_universal/test_gemm_universal_ut_cases_bf16.inc
View File

@@ -1,3 +1,6 @@
// Copyright © Advanced Micro Devices, Inc., or its affiliates.
// SPDX-License-Identifier: MIT
#pragma once
TYPED_TEST(TestGemmUniversal_BF16_MK_KN, SmallM)

3
test/gemm_universal/test_gemm_universal_ut_cases_fp16.inc
View File

@@ -1,3 +1,6 @@
// Copyright © Advanced Micro Devices, Inc., or its affiliates.
// SPDX-License-Identifier: MIT
#pragma once
TYPED_TEST(TestGemmUniversal_FP16_MK_KN, SmallM)

3
test/gemm_universal/test_gemm_universal_ut_cases_fp8.inc
View File

@@ -1,3 +1,6 @@
// Copyright © Advanced Micro Devices, Inc., or its affiliates.
// SPDX-License-Identifier: MIT
#pragma once
TYPED_TEST(TestGemmUniversal_FP8_MK_KN, SmallM)

3
test/gemm_universal_streamk/test_gemm_universal_streamk_ut_cases_bf16.inc
View File

@@ -1,3 +1,6 @@
// Copyright © Advanced Micro Devices, Inc., or its affiliates.
// SPDX-License-Identifier: MIT
#pragma once
TYPED_TEST(TestGemmUniversal_Streamk_BF16_MK_KN, SmallM)

3
test/gemm_universal_streamk/test_gemm_universal_streamk_ut_cases_fp16.inc
View File

@@ -1,3 +1,6 @@
// Copyright © Advanced Micro Devices, Inc., or its affiliates.
// SPDX-License-Identifier: MIT
#pragma once
TYPED_TEST(TestGemmUniversal_Streamk_FP16_MK_KN, SmallM)

3
test/gemm_universal_streamk/test_gemm_universal_streamk_ut_cases_fp8.inc
View File

@@ -1,3 +1,6 @@
// Copyright © Advanced Micro Devices, Inc., or its affiliates.
// SPDX-License-Identifier: MIT
#pragma once
TYPED_TEST(TestGemmUniversal_Streamk_FP8_MK_KN, SmallM)