Commit b24c5f66 authored by Po-Yen, Chen's avatar Po-Yen, Chen
Browse files

Rename 'GridwisePermute' to 'GridwiseCopy'

parent 18ba135f
......@@ -10,7 +10,7 @@
#include "ck/utility/math.hpp"
#include "ck/utility/sequence.hpp"
#include "ck/tensor_operation/gpu/device/device_base.hpp"
#include "ck/tensor_operation/gpu/grid/gridwise_permute.hpp"
#include "ck/tensor_operation/gpu/grid/gridwise_copy.hpp"
#include "ck/tensor_description/tensor_descriptor_helper.hpp"
#include "ck/host_utility/kernel_launch.hpp"
......@@ -166,14 +166,14 @@ struct DevicePermute : detail::DevicePermuteBase<DevicePermute<InDataType,
using InGrid1dDesc = decltype(MakeDescriptor_N_H_W({1, 1}, {1, 1}, 1, 1));
using OutGrid1dDesc = decltype(MakeDescriptor_N_H_W({1, 1}, {1, 1}, 1, 1));
using GridwisePermute = GridwisePermute<InGrid1dDesc,
OutGrid1dDesc,
InDataTypePointer,
OutDataTypePointer,
ElementwiseOperation,
MPerThread,
InScalarPerVector,
OutScalarPerVector>;
using GridwiseCopy = GridwiseCopy<InGrid1dDesc,
OutGrid1dDesc,
InDataTypePointer,
OutDataTypePointer,
ElementwiseOperation,
MPerThread,
InScalarPerVector,
OutScalarPerVector>;
struct Argument : public BaseArgument
{
......@@ -218,7 +218,7 @@ struct DevicePermute : detail::DevicePermuteBase<DevicePermute<InDataType,
{
static float Run(const Argument& arg, const StreamConfig& stream_config = StreamConfig{})
{
const auto kernel = kernel_permute<GridwisePermute,
const auto kernel = kernel_nd_copy<GridwiseCopy,
InGrid1dDesc,
OutGrid1dDesc,
InDataTypePointer,
......
......@@ -10,19 +10,19 @@
namespace ck {
template <typename GridwisePermuteFunctor,
template <typename GridwiseCopyFunctor,
typename InGrid1dDesc,
typename OutGrid1dDesc,
typename InDataTypePointer,
typename OutDataTypePointer,
typename ElementwiseOperation>
__global__ void kernel_permute(const InGrid1dDesc in_grid_1d_desc,
__global__ void kernel_nd_copy(const InGrid1dDesc in_grid_1d_desc,
const OutGrid1dDesc out_grid_1d_desc,
const InDataTypePointer p_in_global,
const OutDataTypePointer p_out_global,
const ElementwiseOperation elementwise_op)
{
GridwisePermuteFunctor::Run(
GridwiseCopyFunctor::Run(
in_grid_1d_desc, out_grid_1d_desc, p_in_global, p_out_global, elementwise_op);
}
......@@ -34,7 +34,7 @@ template <typename InGrid1dDesc,
index_t MPerThread,
index_t InScalarPerVector,
index_t OutScalarPerVector>
struct GridwisePermute
struct GridwiseCopy
{
static constexpr auto I0 = Number<0>{};
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment