From 45966d878ad425750f213cb5808e6bb49ea8b4f4 Mon Sep 17 00:00:00 2001 From: Jesse Benson Date: Wed, 18 Nov 2020 14:08:39 -0800 Subject: [PATCH] Code review feedback --- orttraining/orttraining/training_ops/cuda/math/isfinite.cc | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/orttraining/orttraining/training_ops/cuda/math/isfinite.cc b/orttraining/orttraining/training_ops/cuda/math/isfinite.cc index e0ecf15dd2..c341cf64fe 100644 --- a/orttraining/orttraining/training_ops/cuda/math/isfinite.cc +++ b/orttraining/orttraining/training_ops/cuda/math/isfinite.cc @@ -55,10 +55,10 @@ Status IsAllFiniteOp::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::MappedType*>(output.template MutableData()); + auto* output_data = reinterpret_cast::MappedType*>(output.template MutableData()); CUDA_RETURN_IF_ERROR(cudaMemsetAsync(output_data, int(true), sizeof(bool))); std::vector> grouped_tensor_pointers(total_tensor_count);