From e074c7ccb360d22232a7071f7cf0a4241de8465b Mon Sep 17 00:00:00 2001 From: guyang3532 Date: Wed, 8 Nov 2023 07:34:05 +0000 Subject: [PATCH] fix comments --- .../cuda/tensor/flatten_and_unpad_impl.cu | 14 +++++++------- .../cuda/tensor/pad_and_unflatten_impl.cu | 10 +++++----- 2 files changed, 12 insertions(+), 12 deletions(-) diff --git a/orttraining/orttraining/training_ops/cuda/tensor/flatten_and_unpad_impl.cu b/orttraining/orttraining/training_ops/cuda/tensor/flatten_and_unpad_impl.cu index cd2eb98d5e0d5..2091a7082ee79 100644 --- a/orttraining/orttraining/training_ops/cuda/tensor/flatten_and_unpad_impl.cu +++ b/orttraining/orttraining/training_ops/cuda/tensor/flatten_and_unpad_impl.cu @@ -61,7 +61,7 @@ void FlattenAndUnpadImpl(cudaStream_t stream, output_data); } -#define FLATTEN_AND_UNPAD_FROM_MASK_IMPL(T) \ +#define FLATTEN_AND_UNPAD_IMPL(T) \ template void FlattenAndUnpadImpl(cudaStream_t stream, \ const int64_t total_element_count, \ const fast_divmod output_element_stride_fdm, \ @@ -70,12 +70,12 @@ void FlattenAndUnpadImpl(cudaStream_t stream, const int64_t* indices_data, \ T* output_data); -FLATTEN_AND_UNPAD_FROM_MASK_IMPL(float) -FLATTEN_AND_UNPAD_FROM_MASK_IMPL(double) -FLATTEN_AND_UNPAD_FROM_MASK_IMPL(half) -FLATTEN_AND_UNPAD_FROM_MASK_IMPL(BFloat16) -FLATTEN_AND_UNPAD_FROM_MASK_IMPL(int32_t) -FLATTEN_AND_UNPAD_FROM_MASK_IMPL(int64_t) +FLATTEN_AND_UNPAD_IMPL(float) +FLATTEN_AND_UNPAD_IMPL(double) +FLATTEN_AND_UNPAD_IMPL(half) +FLATTEN_AND_UNPAD_IMPL(BFloat16) +FLATTEN_AND_UNPAD_IMPL(int32_t) +FLATTEN_AND_UNPAD_IMPL(int64_t) #undef FLATTEN_AND_UNPAD_FROM_MASK_IMPL diff --git a/orttraining/orttraining/training_ops/cuda/tensor/pad_and_unflatten_impl.cu b/orttraining/orttraining/training_ops/cuda/tensor/pad_and_unflatten_impl.cu index 71340264a6ab5..e96770f974bf0 100644 --- a/orttraining/orttraining/training_ops/cuda/tensor/pad_and_unflatten_impl.cu +++ b/orttraining/orttraining/training_ops/cuda/tensor/pad_and_unflatten_impl.cu @@ -61,7 +61,7 @@ void PadAndUnflattenImpl(cudaStream_t stream, output_data); } -#define PAD_AND_UNFLATTEN_FROM_MASK_IMPL(T) \ +#define PAD_AND_UNFLATTEN_IMPL(T) \ template void PadAndUnflattenImpl(cudaStream_t stream, \ const int64_t total_element_count, \ const fast_divmod output_element_stride_fdm, \ @@ -70,10 +70,10 @@ void PadAndUnflattenImpl(cudaStream_t stream, const int64_t* indices_data, \ T* output_data); -PAD_AND_UNFLATTEN_FROM_MASK_IMPL(float) -PAD_AND_UNFLATTEN_FROM_MASK_IMPL(double) -PAD_AND_UNFLATTEN_FROM_MASK_IMPL(half) -PAD_AND_UNFLATTEN_FROM_MASK_IMPL(BFloat16) +PAD_AND_UNFLATTEN_IMPL(float) +PAD_AND_UNFLATTEN_IMPL(double) +PAD_AND_UNFLATTEN_IMPL(half) +PAD_AND_UNFLATTEN_IMPL(BFloat16) #undef PAD_AND_UNFLATTEN_FROM_MASK_IMPL