mirror of
https://github.com/ROCm/composable_kernel.git
synced 2026-05-14 18:17:44 +00:00
ec7bb1f840
* updating codegen build for MIOpen access: adding .cmake for codegen component
* updating CMake
* adding in header guards for some headers due to issues with hiprtc compilation in MIOpen
* some more header guards
* putting env file in header guard
* cleaning up some includes
* updated types file for hiprtc purposes
* fixed types file: bit-wise/memcpy issue
* updating multiple utility files to deal with standard header inclusion for hiprtc
* added some more header guards in the utility files, replacing some standard header functionality
* added some more header guards
* fixing some conflicts in utility files, another round of header guards
* fixing errors in data type file
* resolved conflict errors in a few utility files
* added header guards/replicated functionality in device files
* resolved issues with standard headers in device files: device_base and device_grouped_conv_fwd_multiple_abd
* resolved issues with standard headers in device files: device_base.hpp, device_grouped_conv_fwd_multiple_abd.hpp, device_grouped_conv_fwd_multiple_abd_xdl_cshuffle.hpp
* added header guards for gridwise gemm files: gridwise_gemm_multiple_abd_xdl_cshuffle.hpp and gridwise_gemm_multiple_d_xdl_cshuffle.hpp
* fixed issue with numerics header, removed from transform_conv_fwd_to_gemm and added to device_column_to_image_impl, device_grouped_conv_fwd_multiple_abd_xdl_cshuffle, device_grouped_conv_fwd_multiple_abd_xdl_cshuffle_v3, device_image_to_column_impl
* replaced standard header usage and added header guards in block to ctile map and gridwise_gemm_pipeline_selector
* resolved errors in device_gemm_xdl_splitk_c_shuffle files in regards to replacement of standard headers in previous commit
* added replicated functionality for standard header methods in utility files
* replaced standard header functionality in threadwise tensor slice transfer files and added header guards in element_wise_operation.hpp
* temp fix for namespace error in MIOpen
* remove standard header usage in codegen device op
* removed standard header usage in elementwise files, resolved namespace errors
* formatting fix
* changed codegen argument to ON for testing
* temporarily removing codegen compiler flag for testing purposes
* added codegen flag again, set default to ON
* set codegen flag default back to OFF
* replaced enable_if_t standard header usage in data_type.hpp
* added some debug prints to pinpoint issues in MIOpen
* added print outs to debug in MIOpen
* removed debug print outs from device op
* resolved stdexcept include error
* formatting fix
* adding includes to new fp8 file to resolve ck::enable_if_t errors
* made changes to amd_wave_read_first_lane
* updated functionality in type utility file
* fixed end of file issue
* resovled errors in type utility file, added functionality to array utility file
* fixed standard header usage replication in data_type file, resolves error with failing examples on navi3x
* formatting fix
* replaced standard header usage in amd_ck_fp8 file
* added include to random_gen file
* removed and replicated standard header usage from data_type and type_convert files for fp8 changes
* replicated standard unsigned integer types in random_gen
* resolved comments from review: put calls to reinterpret_cast for size_t in header guards
* updated/added copyright headers
* removed duplicate header
* fixed typo in header guard
* updated copyright headers
---------
Co-authored-by: Illia Silin <98187287+illsilin@users.noreply.github.com>
[ROCm/composable_kernel commit: 2e3183af4f]
66 lines
1.9 KiB
C++
66 lines
1.9 KiB
C++
// SPDX-License-Identifier: MIT
|
|
// Copyright (c) 2018-2025, Advanced Micro Devices, Inc. All rights reserved.
|
|
|
|
#ifndef CK_FUNCTIONAL4_HPP
|
|
#define CK_FUNCTIONAL4_HPP
|
|
|
|
#include "sequence.hpp"
|
|
#include "tuple.hpp"
|
|
#include "array.hpp"
|
|
|
|
namespace ck {
|
|
|
|
namespace detail {
|
|
|
|
template <typename Indices>
|
|
struct unpack_impl;
|
|
|
|
template <index_t... Is>
|
|
struct unpack_impl<Sequence<Is...>>
|
|
{
|
|
template <typename F, typename X>
|
|
__host__ __device__ constexpr auto operator()(F&& f, X&& x) const
|
|
{
|
|
return ck::forward<F>(f)(ck::forward<X>(x).At(Number<Is>{})...);
|
|
}
|
|
};
|
|
|
|
template <typename Seq0, typename Seq1>
|
|
struct unpack2_impl;
|
|
|
|
// TODO: remove this, after properly implementing unpack that takes any number of containers
|
|
template <index_t... Is, index_t... Js>
|
|
struct unpack2_impl<Sequence<Is...>, Sequence<Js...>>
|
|
{
|
|
template <typename F, typename X, typename Y>
|
|
__host__ __device__ constexpr auto operator()(F&& f, X&& x, Y&& y) const
|
|
{
|
|
return ck::forward<F>(f)(ck::forward<X>(x).At(Number<Is>{})...,
|
|
ck::forward<Y>(y).At(Number<Js>{})...);
|
|
}
|
|
};
|
|
|
|
} // namespace detail
|
|
|
|
template <typename F, typename X>
|
|
__host__ __device__ constexpr auto unpack(F&& f, X&& x)
|
|
{
|
|
using X_ = remove_reference_t<X>;
|
|
return detail::unpack_impl<typename arithmetic_sequence_gen<0, X_::Size(), 1>::type>{}(
|
|
ck::forward<F>(f), ck::forward<X>(x));
|
|
}
|
|
|
|
// TODO: properly implement unpack that takes any number of containers
|
|
template <typename F, typename X, typename Y>
|
|
__host__ __device__ constexpr auto unpack2(F&& f, X&& x, Y&& y)
|
|
{
|
|
using X_ = remove_reference_t<X>;
|
|
using Y_ = remove_reference_t<Y>;
|
|
return detail::unpack2_impl<typename arithmetic_sequence_gen<0, X_::Size(), 1>::type,
|
|
typename arithmetic_sequence_gen<0, Y_::Size(), 1>::type>{}(
|
|
ck::forward<F>(f), ck::forward<X>(x), ck::forward<Y>(y));
|
|
}
|
|
|
|
} // namespace ck
|
|
#endif
|