saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-07-29 20:14:01 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Revert _SliceKernel cuda implementation

Browse source
This commit is contained in:
Thiago Crepaldi 2020-04-01 14:28:17 -07:00
parent 28ff88ce52
commit 0b1e3f1e10
1 changed files with 7 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

11
onnxruntime/core/providers/cuda/tensor/slice_impl.cu
View file

@ -29,12 +29,15 @@ __global__ void _SliceKernel(const int32_t dimension_count,
break;
}
output_strides[dim].divmod(value, div, mod);
input_index += (starts[dim] + div * steps[dim]) * input_strides[dim];
output_strides.data_[dim].divmod(value, div, mod);
input_index += (starts.data_[dim] + div * steps.data_[dim]) * input_strides.data_[dim];
value = mod;
}
input_index += starts[dim] + mod * steps[dim];
output_data[id] = input_data[input_index];
input_index += starts.data_[dim] + mod * steps.data_[dim];
if (is_grad)
output_data[input_index] = input_data[id];
else
output_data[id] = input_data[input_index];
}
Status SliceImpl(const size_t element_size,