mirror of
https://github.com/ROCm/composable_kernel.git
synced 2026-05-13 09:45:56 +00:00
8eff4d62b6
* Format * Format * Format * Remove const * Use the right template * Format * Format * add row/col instances * Add missing file * fixed * Format * Updates * Format * fixed rrr layout * Format * Update test and embed modules * Restore older version * Update year * Set -fPIC * Format * Use double for isnan * rename host folder to codegen + minor fix * add codegen CI test * add option to build components without building CK * fix the groovy syntax * fix typo * use the correct function for the codegen stage --------- Co-authored-by: Jing Zhang <jizha@amd.com> Co-authored-by: Illia Silin <98187287+illsilin@users.noreply.github.com> Co-authored-by: illsilin <Illia.Silin@amd.com>
63 lines
1.6 KiB
C++
63 lines
1.6 KiB
C++
#ifndef GUARD_HOST_TEST_RTC_INCLUDE_RTC_KERNEL
|
|
#define GUARD_HOST_TEST_RTC_INCLUDE_RTC_KERNEL
|
|
|
|
#include <hip/hip_runtime_api.h>
|
|
#include <memory>
|
|
#include <string>
|
|
#include <vector>
|
|
|
|
namespace rtc {
|
|
|
|
struct kernel_argument
|
|
{
|
|
template <class T,
|
|
class U = std::remove_reference_t<T>,
|
|
class = std::enable_if_t<not std::is_base_of<kernel_argument, T>{}>>
|
|
kernel_argument(T&& x) : size(sizeof(U)), align(alignof(U)), data(&x) // NOLINT
|
|
{
|
|
}
|
|
std::size_t size;
|
|
std::size_t align;
|
|
void* data;
|
|
};
|
|
|
|
std::vector<char> pack_args(const std::vector<kernel_argument>& args);
|
|
|
|
struct kernel_impl;
|
|
|
|
struct kernel
|
|
{
|
|
kernel() = default;
|
|
kernel(const char* image, const std::string& name);
|
|
template <class T>
|
|
kernel(const std::vector<T>& image, const std::string& name)
|
|
: kernel(reinterpret_cast<const char*>(image.data()), name)
|
|
{
|
|
static_assert(sizeof(T) == 1, "Only byte types");
|
|
}
|
|
|
|
void launch(hipStream_t stream,
|
|
std::size_t global,
|
|
std::size_t local,
|
|
const std::vector<kernel_argument>& args) const;
|
|
|
|
void launch(hipStream_t stream,
|
|
std::size_t global,
|
|
std::size_t local,
|
|
std::vector<void*> args) const;
|
|
|
|
template <class... Ts>
|
|
auto launch(hipStream_t stream, std::size_t global, std::size_t local, Ts... zs) const
|
|
{
|
|
return [=](auto&&... xs) {
|
|
launch(stream, global, local, std::vector<kernel_argument>{xs...}, zs...);
|
|
};
|
|
}
|
|
|
|
private:
|
|
std::shared_ptr<kernel_impl> impl;
|
|
};
|
|
} // namespace rtc
|
|
|
|
#endif
|