Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
gaoqiong
composable_kernel_ROCM
Commits
d480a5a6
Unverified
Commit
d480a5a6
authored
Feb 03, 2025
by
Max Podkorytov
Committed by
GitHub
Feb 03, 2025
Browse files
Merge branch 'develop' into ck-flex
parents
bca939ce
9c5b2f39
Changes
94
Hide whitespace changes
Inline
Side-by-side
Showing
20 changed files
with
58 additions
and
4 deletions
+58
-4
CMakeLists.txt
CMakeLists.txt
+1
-1
Jenkinsfile
Jenkinsfile
+2
-2
codegen/driver/main.cpp
codegen/driver/main.cpp
+2
-0
codegen/src/headers.cpp
codegen/src/headers.cpp
+3
-0
codegen/src/types.cpp
codegen/src/types.cpp
+3
-0
codegen/test/gemm_multiple_d.cpp
codegen/test/gemm_multiple_d.cpp
+3
-0
codegen/test/grouped_conv_fwd_multiple_d_v1.cpp
codegen/test/grouped_conv_fwd_multiple_d_v1.cpp
+3
-0
codegen/test/grouped_conv_fwd_multiple_d_v2.cpp
codegen/test/grouped_conv_fwd_multiple_d_v2.cpp
+3
-0
codegen/test/grouped_conv_fwd_multiple_d_v3.cpp
codegen/test/grouped_conv_fwd_multiple_d_v3.cpp
+3
-0
codegen/test/grouped_conv_fwd_multiple_d_v4.cpp
codegen/test/grouped_conv_fwd_multiple_d_v4.cpp
+3
-0
codegen/test/include/common.hpp
codegen/test/include/common.hpp
+3
-0
codegen/test/rtc/include/rtc/compile_kernel.hpp
codegen/test/rtc/include/rtc/compile_kernel.hpp
+3
-0
codegen/test/rtc/include/rtc/hip.hpp
codegen/test/rtc/include/rtc/hip.hpp
+4
-1
codegen/test/rtc/include/rtc/kernel.hpp
codegen/test/rtc/include/rtc/kernel.hpp
+3
-0
codegen/test/rtc/include/rtc/manage_ptr.hpp
codegen/test/rtc/include/rtc/manage_ptr.hpp
+3
-0
codegen/test/rtc/include/rtc/tmp_dir.hpp
codegen/test/rtc/include/rtc/tmp_dir.hpp
+3
-0
codegen/test/rtc/src/compile_kernel.cpp
codegen/test/rtc/src/compile_kernel.cpp
+3
-0
codegen/test/rtc/src/hip.cpp
codegen/test/rtc/src/hip.cpp
+3
-0
codegen/test/rtc/src/kernel.cpp
codegen/test/rtc/src/kernel.cpp
+4
-0
codegen/test/rtc/src/tmp_dir.cpp
codegen/test/rtc/src/tmp_dir.cpp
+3
-0
No files found.
CMakeLists.txt
View file @
d480a5a6
...
...
@@ -103,7 +103,7 @@ if(DPP_KERNELS)
endif
()
option
(
CK_USE_CODEGEN
"Enable codegen library"
OFF
)
if
(
CK_USE_CODEGEN
)
add_definitions
(
-DCK_USE_CODEGEN
)
add_definitions
(
-DCK_USE_CODEGEN
)
endif
()
option
(
CK_TIME_KERNEL
"Enable kernel time tracking"
ON
)
...
...
Jenkinsfile
View file @
d480a5a6
...
...
@@ -795,8 +795,8 @@ pipeline {
description:
"Run the ck_tile FMHA tests (default: OFF)"
)
booleanParam
(
name:
"RUN_CK_TILE_GEMM_TESTS"
,
defaultValue:
fals
e
,
description:
"Run the ck_tile GEMM tests (default: O
FF
)"
)
defaultValue:
tru
e
,
description:
"Run the ck_tile GEMM tests (default: O
N
)"
)
booleanParam
(
name:
"BUILD_INSTANCES_ONLY"
,
defaultValue:
false
,
...
...
codegen/driver/main.cpp
View file @
d480a5a6
// SPDX-License-Identifier: MIT
// Copyright (c) 2024-2025, Advanced Micro Devices, Inc. All rights reserved.
#include <functional>
#include <iostream>
...
...
codegen/src/headers.cpp
View file @
d480a5a6
// SPDX-License-Identifier: MIT
// Copyright (c) 2024-2025, Advanced Micro Devices, Inc. All rights reserved.
#include "ck/host/headers.hpp"
#include "ck_headers.hpp"
...
...
codegen/src/types.cpp
View file @
d480a5a6
// SPDX-License-Identifier: MIT
// Copyright (c) 2024-2025, Advanced Micro Devices, Inc. All rights reserved.
#include "ck/host/types.hpp"
#include "ck/host/stringutils.hpp"
#include <algorithm>
...
...
codegen/test/gemm_multiple_d.cpp
View file @
d480a5a6
// SPDX-License-Identifier: MIT
// Copyright (c) 2024-2025, Advanced Micro Devices, Inc. All rights reserved.
#include "ck/host/device_gemm_multiple_d/problem.hpp"
#include "ck/host/device_gemm_multiple_d/operation.hpp"
#include "ck/host/headers.hpp"
...
...
codegen/test/grouped_conv_fwd_multiple_d_v1.cpp
View file @
d480a5a6
// SPDX-License-Identifier: MIT
// Copyright (c) 2024-2025, Advanced Micro Devices, Inc. All rights reserved.
#include "ck/host/device_grouped_conv_fwd_multiple_d/conv_fwd_op.hpp"
#include "ck/host/device_grouped_conv_fwd_multiple_d/conv_fwd_problem.hpp"
#include "ck/host/headers.hpp"
...
...
codegen/test/grouped_conv_fwd_multiple_d_v2.cpp
View file @
d480a5a6
// SPDX-License-Identifier: MIT
// Copyright (c) 2024-2025, Advanced Micro Devices, Inc. All rights reserved.
#include "ck/host/device_grouped_conv_fwd_multiple_d/conv_fwd_op.hpp"
#include "ck/host/device_grouped_conv_fwd_multiple_d/conv_fwd_problem.hpp"
#include "ck/host/headers.hpp"
...
...
codegen/test/grouped_conv_fwd_multiple_d_v3.cpp
View file @
d480a5a6
// SPDX-License-Identifier: MIT
// Copyright (c) 2024-2025, Advanced Micro Devices, Inc. All rights reserved.
#include "ck/host/device_grouped_conv_fwd_multiple_d/conv_fwd_op.hpp"
#include "ck/host/device_grouped_conv_fwd_multiple_d/conv_fwd_problem.hpp"
#include "ck/host/headers.hpp"
...
...
codegen/test/grouped_conv_fwd_multiple_d_v4.cpp
View file @
d480a5a6
// SPDX-License-Identifier: MIT
// Copyright (c) 2024-2025, Advanced Micro Devices, Inc. All rights reserved.
#include "ck/host/device_grouped_conv_fwd_multiple_d/conv_fwd_op.hpp"
#include "ck/host/device_grouped_conv_fwd_multiple_d/conv_fwd_problem.hpp"
#include "ck/host/headers.hpp"
...
...
codegen/test/include/common.hpp
View file @
d480a5a6
// SPDX-License-Identifier: MIT
// Copyright (c) 2024-2025, Advanced Micro Devices, Inc. All rights reserved.
#pragma once
#include <algorithm>
#include <cmath>
...
...
codegen/test/rtc/include/rtc/compile_kernel.hpp
View file @
d480a5a6
// SPDX-License-Identifier: MIT
// Copyright (c) 2024-2025, Advanced Micro Devices, Inc. All rights reserved.
#ifndef GUARD_HOST_TEST_RTC_INCLUDE_RTC_COMPILE_KERNEL
#define GUARD_HOST_TEST_RTC_INCLUDE_RTC_COMPILE_KERNEL
...
...
codegen/test/rtc/include/rtc/hip.hpp
View file @
d480a5a6
// SPDX-License-Identifier: MIT
// Copyright (c) 2024-2025, Advanced Micro Devices, Inc. All rights reserved.
#ifndef GUARD_HOST_TEST_RTC_INCLUDE_RTC_HIP
#define GUARD_HOST_TEST_RTC_INCLUDE_RTC_HIP
#include <hip/hip_runtime_api.h>
#include <memory>
#include <string>
#include <stdexcept>
#include <string>
namespace
rtc
{
...
...
codegen/test/rtc/include/rtc/kernel.hpp
View file @
d480a5a6
// SPDX-License-Identifier: MIT
// Copyright (c) 2024-2025, Advanced Micro Devices, Inc. All rights reserved.
#ifndef GUARD_HOST_TEST_RTC_INCLUDE_RTC_KERNEL
#define GUARD_HOST_TEST_RTC_INCLUDE_RTC_KERNEL
...
...
codegen/test/rtc/include/rtc/manage_ptr.hpp
View file @
d480a5a6
// SPDX-License-Identifier: MIT
// Copyright (c) 2024-2025, Advanced Micro Devices, Inc. All rights reserved.
#ifndef GUARD_HOST_TEST_RTC_INCLUDE_RTC_MANAGE_POINTER
#define GUARD_HOST_TEST_RTC_INCLUDE_RTC_MANAGE_POINTER
...
...
codegen/test/rtc/include/rtc/tmp_dir.hpp
View file @
d480a5a6
// SPDX-License-Identifier: MIT
// Copyright (c) 2024-2025, Advanced Micro Devices, Inc. All rights reserved.
#ifndef GUARD_HOST_TEST_RTC_INCLUDE_RTC_TMP_DIR
#define GUARD_HOST_TEST_RTC_INCLUDE_RTC_TMP_DIR
...
...
codegen/test/rtc/src/compile_kernel.cpp
View file @
d480a5a6
// SPDX-License-Identifier: MIT
// Copyright (c) 2024-2025, Advanced Micro Devices, Inc. All rights reserved.
#include <rtc/hip.hpp>
#include <rtc/compile_kernel.hpp>
#include <rtc/tmp_dir.hpp>
...
...
codegen/test/rtc/src/hip.cpp
View file @
d480a5a6
// SPDX-License-Identifier: MIT
// Copyright (c) 2024-2025, Advanced Micro Devices, Inc. All rights reserved.
#include <rtc/hip.hpp>
#include <rtc/manage_ptr.hpp>
#include <stdexcept>
...
...
codegen/test/rtc/src/kernel.cpp
View file @
d480a5a6
// SPDX-License-Identifier: MIT
// Copyright (c) 2024-2025, Advanced Micro Devices, Inc. All rights reserved.
#include <rtc/kernel.hpp>
#include <rtc/manage_ptr.hpp>
#include <rtc/hip.hpp>
#include <stdexcept>
#include <cassert>
// extern declare the function since hip/hip_ext.h header is broken
...
...
codegen/test/rtc/src/tmp_dir.cpp
View file @
d480a5a6
// SPDX-License-Identifier: MIT
// Copyright (c) 2024-2025, Advanced Micro Devices, Inc. All rights reserved.
#include <rtc/tmp_dir.hpp>
#include <algorithm>
#include <random>
...
...
Prev
1
2
3
4
5
Next
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment