Skip to content

GitLab

"src/node/vscode:/vscode.git/clone" did not exist on "ac81d7c8832b0d0526e34b7408fd18d378b8be27"
Commit be56fdef authored by Astha Rai's avatar Astha Rai
Browse files

added dimensions for example file

parent 08848bb6
Show whitespace changes
Inline Side-by-side
Showing with 4 additions and 3 deletions
include/ck/tensor_operation/gpu/device/device_elementwise_2d.hpp
View file @ be56fdef
......@@ -96,7 +96,8 @@ struct DeviceElementwise
const auto desc = make_naive_tensor_descriptor(tupleOfShape, tupleOfStride);
constexpr auto mDimIds = typename arithmetic_sequence_gen<0, NumDim_m, 1>::type();
constexpr auto nDimIds = typename arithmetic_sequence_gen< NumDim_m, NumDim_m + NumDim_n, 1>::type();
constexpr auto nDimIds =
typename arithmetic_sequence_gen<NumDim_m, NumDim_m + NumDim_n, 1>::type();
const auto mLengths = get_container_subset(tupleOfShape, mDimIds);
const auto nLengths = get_container_subset(tupleOfShape, nDimIds);
......@@ -136,7 +137,7 @@ struct DeviceElementwise
using OutGrid2dDescTuple = decltype(GenerateInOutGrid2dDescTuple(Number<NumOutput>{}));
using InGrid2dDescTuple = decltype(GenerateInOutGrid2dDescTuple(Number<NumInput>{}));
//using OutGrid2dDescTuple = decltype(GenerateInOutGrid2dDescTuple(Number<NumOutput>{}));
// using OutGrid2dDescTuple = decltype(GenerateInOutGrid2dDescTuple(Number<NumOutput>{}));
using GridwiseElementwise = GridwiseElementwise_2D<InGrid2dDescTuple,
OutGrid2dDescTuple,
......
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