Commit 53468445 authored by Catheriany's avatar Catheriany
Browse files

issue/282: 算子贴错导致推理问题修复

parent af8bdb43
...@@ -18,7 +18,7 @@ INFINIOP_MACA_KERNEL causalSoftmax( ...@@ -18,7 +18,7 @@ INFINIOP_MACA_KERNEL causalSoftmax(
// [Reduce] Find max value in each row and store in shared memory // [Reduce] Find max value in each row and store in shared memory
__shared__ Tdata max_; __shared__ Tdata max_;
Tdata max_0 = op::common_maca::reduce_op::max<BLOCK_SIZE, Tdata>(x, width); Tdata max_0 = op::common_maca::reduce_op::max<BLOCK_SIZE, Tdata>(x, width - height + 1 + blockIdx.x);
if (threadIdx.x == 0) { if (threadIdx.x == 0) {
max_ = max_0; max_ = max_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