"docs/source/vscode:/vscode.git/clone" did not exist on "d93ca268938940839118fdb8d0cf4c1ca7a9fead"
Commit b24c5f66 authored by Po-Yen, Chen's avatar Po-Yen, Chen
Browse files

Rename 'GridwisePermute' to 'GridwiseCopy'

parent 18ba135f
...@@ -10,7 +10,7 @@ ...@@ -10,7 +10,7 @@
#include "ck/utility/math.hpp" #include "ck/utility/math.hpp"
#include "ck/utility/sequence.hpp" #include "ck/utility/sequence.hpp"
#include "ck/tensor_operation/gpu/device/device_base.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/tensor_description/tensor_descriptor_helper.hpp"
#include "ck/host_utility/kernel_launch.hpp" #include "ck/host_utility/kernel_launch.hpp"
...@@ -166,14 +166,14 @@ struct DevicePermute : detail::DevicePermuteBase<DevicePermute<InDataType, ...@@ -166,14 +166,14 @@ struct DevicePermute : detail::DevicePermuteBase<DevicePermute<InDataType,
using InGrid1dDesc = decltype(MakeDescriptor_N_H_W({1, 1}, {1, 1}, 1, 1)); 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 OutGrid1dDesc = decltype(MakeDescriptor_N_H_W({1, 1}, {1, 1}, 1, 1));
using GridwisePermute = GridwisePermute<InGrid1dDesc, using GridwiseCopy = GridwiseCopy<InGrid1dDesc,
OutGrid1dDesc, OutGrid1dDesc,
InDataTypePointer, InDataTypePointer,
OutDataTypePointer, OutDataTypePointer,
ElementwiseOperation, ElementwiseOperation,
MPerThread, MPerThread,
InScalarPerVector, InScalarPerVector,
OutScalarPerVector>; OutScalarPerVector>;
struct Argument : public BaseArgument struct Argument : public BaseArgument
{ {
...@@ -218,7 +218,7 @@ struct DevicePermute : detail::DevicePermuteBase<DevicePermute<InDataType, ...@@ -218,7 +218,7 @@ struct DevicePermute : detail::DevicePermuteBase<DevicePermute<InDataType,
{ {
static float Run(const Argument& arg, const StreamConfig& stream_config = StreamConfig{}) 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, InGrid1dDesc,
OutGrid1dDesc, OutGrid1dDesc,
InDataTypePointer, InDataTypePointer,
......
...@@ -10,19 +10,19 @@ ...@@ -10,19 +10,19 @@
namespace ck { namespace ck {
template <typename GridwisePermuteFunctor, template <typename GridwiseCopyFunctor,
typename InGrid1dDesc, typename InGrid1dDesc,
typename OutGrid1dDesc, typename OutGrid1dDesc,
typename InDataTypePointer, typename InDataTypePointer,
typename OutDataTypePointer, typename OutDataTypePointer,
typename ElementwiseOperation> 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 OutGrid1dDesc out_grid_1d_desc,
const InDataTypePointer p_in_global, const InDataTypePointer p_in_global,
const OutDataTypePointer p_out_global, const OutDataTypePointer p_out_global,
const ElementwiseOperation elementwise_op) const ElementwiseOperation elementwise_op)
{ {
GridwisePermuteFunctor::Run( GridwiseCopyFunctor::Run(
in_grid_1d_desc, out_grid_1d_desc, p_in_global, p_out_global, elementwise_op); in_grid_1d_desc, out_grid_1d_desc, p_in_global, p_out_global, elementwise_op);
} }
...@@ -34,7 +34,7 @@ template <typename InGrid1dDesc, ...@@ -34,7 +34,7 @@ template <typename InGrid1dDesc,
index_t MPerThread, index_t MPerThread,
index_t InScalarPerVector, index_t InScalarPerVector,
index_t OutScalarPerVector> index_t OutScalarPerVector>
struct GridwisePermute struct GridwiseCopy
{ {
static constexpr auto I0 = Number<0>{}; 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