fix comments

microsoft · Nov 8, 2023 · e074c7c · e074c7c
1 parent 8c7ce15
commit e074c7c
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 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
@@ -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<T>(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
@@ -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<T>(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