Skip to content

GitLab

"...composable_kernel.git" did not exist on "0c51a35ea8a60adb8feb2fc7da876ea45c272730"
Commit 4726df77 authored by rocking's avatar rocking
Browse files

Add comment

parent e9ca131c
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 0 deletions
include/ck/tensor_operation/gpu/device/impl/device_index_pool_bwd_impl.hpp
View file @ 4726df77
...@@ -281,6 +281,8 @@ struct DeviceIndexPoolBwdImpl : public DeviceIndexPoolBwd<DOutDataType, IndexDat ...@@ -281,6 +281,8 @@ struct DeviceIndexPoolBwdImpl : public DeviceIndexPoolBwd<DOutDataType, IndexDat
std::vector<ck::index_t> window_lengths, std::vector<ck::index_t> window_lengths,
std::vector<ck::index_t> window_strides) override std::vector<ck::index_t> window_strides) override
{ {
// Assume p_dout, p_indices, p_din are packed memory space, dout_length and din_length are
// physical size of the packed tensor
return std::make_unique<Argument>(static_cast<const DOutDataType*>(p_dout), return std::make_unique<Argument>(static_cast<const DOutDataType*>(p_dout),
static_cast<const IndexDataType*>(p_indices), static_cast<const IndexDataType*>(p_indices),
static_cast<DInDataType*>(p_din), static_cast<DInDataType*>(p_din),
......
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