Skip to content

GitLab

Commit 6818b58c authored by rocking's avatar rocking
Browse files

Fix compile error

parent 0277c89e
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 3 deletions
example/19_gemm_softmax/gemm_softmax_xdl_fp16.cpp
View file @ 6818b58c
...@@ -86,9 +86,9 @@ using DeviceGemmInstance = ck::tensor_operation::device::DeviceGemmXdl_C_Shuffle ...@@ -86,9 +86,9 @@ using DeviceGemmInstance = ck::tensor_operation::device::DeviceGemmXdl_C_Shuffle
constexpr int Rank = 2; constexpr int Rank = 2;
constexpr int NumReduceDim = 1; constexpr int NumReduceDim = 1;
constexpr ck::ReduceTensorOp_t ReduceOpId = ck::ReduceTensorOp_t::MAX; constexpr ck::ReduceTensorOp ReduceOpId = ck::ReduceTensorOp::MAX;
constexpr ck::NanPropagation_t NanOpt = ck::NanPropagation_t::PROPAGATE_NAN; constexpr ck::NanPropagation NanOpt = ck::NanPropagation::PROPAGATE_NAN;
constexpr bool PropagateNan = (NanOpt == ck::NanPropagation_t::NOT_PROPAGATE_NAN) ? false : true; constexpr bool PropagateNan = (NanOpt == ck::NanPropagation::NOT_PROPAGATE_NAN) ? false : true;
// constexpr ck::ReduceTensorIndices_t IndicesOpt = ck::ReduceTensorIndices_t::NO_INDICES; // constexpr ck::ReduceTensorIndices_t IndicesOpt = ck::ReduceTensorIndices_t::NO_INDICES;
using ReduceOperation = typename ck::reduce_binary_operator<CDataType, ReduceOpId>::opType; using ReduceOperation = typename ck::reduce_binary_operator<CDataType, ReduceOpId>::opType;
using InElementwiseOperation = using InElementwiseOperation =
......
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