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

Code review feedback

Browse source
This commit is contained in:
Jesse Benson 2020-11-18 14:08:39 -08:00 committed by Jesse Benson
parent 86e30a2db6
commit 45966d878a
1 changed files with 3 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
orttraining/orttraining/training_ops/cuda/math/isfinite.cc
View file

@ -55,10 +55,10 @@ Status IsAllFiniteOp<TSrc>::ComputeInternal(OpKernelContext* context) const {
// Get Input tensor count.
const auto total_tensor_count = context->InputCount();
// Initialize the output to true. GPU kernel will set it to false
// if any value in any tensor is non-finite.
// Initialize the output to true. GPU kernel will set it
// to false if any value in any tensor is non-finite.
Tensor& output = *context->Output(0, {});
auto output_data = reinterpret_cast<ToCudaType<bool>::MappedType*>(output.template MutableData<bool>());
auto* output_data = reinterpret_cast<ToCudaType<bool>::MappedType*>(output.template MutableData<bool>());
CUDA_RETURN_IF_ERROR(cudaMemsetAsync(output_data, int(true), sizeof(bool)));
std::vector<std::vector<void*>> grouped_tensor_pointers(total_tensor_count);