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
Commits
0b11569f
Commit
0b11569f
authored
Jul 01, 2022
by
Chao Liu
Browse files
Merge remote-tracking branch 'origin/develop' into batched_gemm_c_permute
parents
e8d3a0fb
fa9a0a5c
Changes
554
Hide whitespace changes
Inline
Side-by-side
Showing
20 changed files
with
63 additions
and
1 deletion
+63
-1
include/ck/utility/math.hpp
include/ck/utility/math.hpp
+5
-0
include/ck/utility/math_v2.hpp
include/ck/utility/math_v2.hpp
+3
-0
include/ck/utility/multi_index.hpp
include/ck/utility/multi_index.hpp
+3
-0
include/ck/utility/number.hpp
include/ck/utility/number.hpp
+3
-0
include/ck/utility/print.hpp
include/ck/utility/print.hpp
+3
-0
include/ck/utility/reduction_common.hpp
include/ck/utility/reduction_common.hpp
+3
-0
include/ck/utility/reduction_enums.hpp
include/ck/utility/reduction_enums.hpp
+3
-0
include/ck/utility/reduction_functions_accumulate.hpp
include/ck/utility/reduction_functions_accumulate.hpp
+4
-1
include/ck/utility/reduction_operator.hpp
include/ck/utility/reduction_operator.hpp
+3
-0
include/ck/utility/sequence.hpp
include/ck/utility/sequence.hpp
+3
-0
include/ck/utility/sequence_helper.hpp
include/ck/utility/sequence_helper.hpp
+3
-0
include/ck/utility/static_buffer.hpp
include/ck/utility/static_buffer.hpp
+3
-0
include/ck/utility/statically_indexed_array.hpp
include/ck/utility/statically_indexed_array.hpp
+3
-0
include/ck/utility/statically_indexed_array_multi_index.hpp
include/ck/utility/statically_indexed_array_multi_index.hpp
+3
-0
include/ck/utility/synchronization.hpp
include/ck/utility/synchronization.hpp
+3
-0
include/ck/utility/thread_group.hpp
include/ck/utility/thread_group.hpp
+3
-0
include/ck/utility/transpose_vectors.hpp
include/ck/utility/transpose_vectors.hpp
+3
-0
include/ck/utility/tuple.hpp
include/ck/utility/tuple.hpp
+3
-0
include/ck/utility/tuple_helper.hpp
include/ck/utility/tuple_helper.hpp
+3
-0
include/ck/utility/type.hpp
include/ck/utility/type.hpp
+3
-0
No files found.
include/ck/utility/math.hpp
View file @
0b11569f
// SPDX-License-Identifier: MIT
// Copyright (c) 2018-2022, Advanced Micro Devices, Inc. All rights reserved.
#pragma once
#include "ck/ck.hpp"
...
...
@@ -145,6 +148,8 @@ __host__ __device__ constexpr auto min(X x, Ys... ys)
template
<
typename
T
>
__device__
T
exp
(
T
x
);
// TODO: add f16 support using v_exp_f16
template
<
>
__device__
float
exp
<
float
>
(
float
x
)
{
...
...
include/ck/utility/math_v2.hpp
View file @
0b11569f
// SPDX-License-Identifier: MIT
// Copyright (c) 2018-2022, Advanced Micro Devices, Inc. All rights reserved.
#pragma once
#include <cmath>
...
...
include/ck/utility/multi_index.hpp
View file @
0b11569f
// SPDX-License-Identifier: MIT
// Copyright (c) 2018-2022, Advanced Micro Devices, Inc. All rights reserved.
#pragma once
#include "common_header.hpp"
...
...
include/ck/utility/number.hpp
View file @
0b11569f
// SPDX-License-Identifier: MIT
// Copyright (c) 2018-2022, Advanced Micro Devices, Inc. All rights reserved.
#ifndef CK_NUMBER_HPP
#define CK_NUMBER_HPP
...
...
include/ck/utility/print.hpp
View file @
0b11569f
// SPDX-License-Identifier: MIT
// Copyright (c) 2018-2022, Advanced Micro Devices, Inc. All rights reserved.
#ifndef CK_PRINT_HPP
#define CK_PRINT_HPP
...
...
include/ck/utility/reduction_common.hpp
View file @
0b11569f
// SPDX-License-Identifier: MIT
// Copyright (c) 2018-2022, Advanced Micro Devices, Inc. All rights reserved.
#pragma once
#include "ck/utility/reduction_enums.hpp"
...
...
include/ck/utility/reduction_enums.hpp
View file @
0b11569f
// SPDX-License-Identifier: MIT
// Copyright (c) 2018-2022, Advanced Micro Devices, Inc. All rights reserved.
#pragma once
namespace
ck
{
...
...
include/ck/utility/reduction_functions_accumulate.hpp
View file @
0b11569f
// SPDX-License-Identifier: MIT
// Copyright (c) 2018-2022, Advanced Micro Devices, Inc. All rights reserved.
#pragma once
#include "ck/utility/data_type.hpp"
...
...
@@ -14,7 +17,7 @@ struct AccumulateWithNanIgnore
{
__device__
static
inline
void
Calculate
(
AccDataType
&
accuVal
,
AccDataType
currVal
)
{
if
(
!
isnan
(
currVal
))
if
(
!
ck
::
math
::
isnan
(
currVal
))
{
ReduceOperation
{}(
accuVal
,
currVal
);
}
...
...
include/ck/utility/reduction_operator.hpp
View file @
0b11569f
// SPDX-License-Identifier: MIT
// Copyright (c) 2018-2022, Advanced Micro Devices, Inc. All rights reserved.
#pragma once
#include "ck/ck.hpp"
...
...
include/ck/utility/sequence.hpp
View file @
0b11569f
// SPDX-License-Identifier: MIT
// Copyright (c) 2018-2022, Advanced Micro Devices, Inc. All rights reserved.
#pragma once
#include "integral_constant.hpp"
...
...
include/ck/utility/sequence_helper.hpp
View file @
0b11569f
// SPDX-License-Identifier: MIT
// Copyright (c) 2018-2022, Advanced Micro Devices, Inc. All rights reserved.
#ifndef CK_SEQUENCE_HELPER_HPP
#define CK_SEQUENCE_HELPER_HPP
...
...
include/ck/utility/static_buffer.hpp
View file @
0b11569f
// SPDX-License-Identifier: MIT
// Copyright (c) 2018-2022, Advanced Micro Devices, Inc. All rights reserved.
#ifndef CK_STATIC_BUFFER_HPP
#define CK_STATIC_BUFFER_HPP
...
...
include/ck/utility/statically_indexed_array.hpp
View file @
0b11569f
// SPDX-License-Identifier: MIT
// Copyright (c) 2018-2022, Advanced Micro Devices, Inc. All rights reserved.
#ifndef CK_STATICALLY_INDEXED_ARRAY_HPP
#define CK_STATICALLY_INDEXED_ARRAY_HPP
...
...
include/ck/utility/statically_indexed_array_multi_index.hpp
View file @
0b11569f
// SPDX-License-Identifier: MIT
// Copyright (c) 2018-2022, Advanced Micro Devices, Inc. All rights reserved.
#ifndef CK_STATICALLY_INDEXED_ARRAY_MULTI_INDEX_HPP
#define CK_STATICALLY_INDEXED_ARRAY_MULTI_INDEX_HPP
...
...
include/ck/utility/synchronization.hpp
View file @
0b11569f
// SPDX-License-Identifier: MIT
// Copyright (c) 2018-2022, Advanced Micro Devices, Inc. All rights reserved.
#pragma once
#include "ck/ck.hpp"
...
...
include/ck/utility/thread_group.hpp
View file @
0b11569f
// SPDX-License-Identifier: MIT
// Copyright (c) 2018-2022, Advanced Micro Devices, Inc. All rights reserved.
#pragma once
#include "get_id.hpp"
...
...
include/ck/utility/transpose_vectors.hpp
View file @
0b11569f
// SPDX-License-Identifier: MIT
// Copyright (c) 2018-2022, Advanced Micro Devices, Inc. All rights reserved.
#pragma once
#include "ck/ck.hpp"
...
...
include/ck/utility/tuple.hpp
View file @
0b11569f
// SPDX-License-Identifier: MIT
// Copyright (c) 2018-2022, Advanced Micro Devices, Inc. All rights reserved.
#pragma once
#include "integral_constant.hpp"
...
...
include/ck/utility/tuple_helper.hpp
View file @
0b11569f
// SPDX-License-Identifier: MIT
// Copyright (c) 2018-2022, Advanced Micro Devices, Inc. All rights reserved.
#pragma once
#include "functional4.hpp"
...
...
include/ck/utility/type.hpp
View file @
0b11569f
// SPDX-License-Identifier: MIT
// Copyright (c) 2018-2022, Advanced Micro Devices, Inc. All rights reserved.
#pragma once
#include "ck/ck.hpp"
...
...
Prev
1
…
8
9
10
11
12
13
14
15
16
…
28
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