From 1a37190c0e25a6e31c3e416052b38c34b9f6c405 Mon Sep 17 00:00:00 2001 From: cyy Date: Sun, 28 Apr 2024 09:38:59 +0800 Subject: [PATCH] Remove unused comments --- torch/csrc/distributed/c10d/quantization/quantization.cpp | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) diff --git a/torch/csrc/distributed/c10d/quantization/quantization.cpp b/torch/csrc/distributed/c10d/quantization/quantization.cpp index 8a6958eb002af..c4c250a00b8f1 100644 --- a/torch/csrc/distributed/c10d/quantization/quantization.cpp +++ b/torch/csrc/distributed/c10d/quantization/quantization.cpp @@ -49,8 +49,7 @@ at::Tensor _float_to_bfloat16_cpu(const at::Tensor& input) { const auto input_sizes = input.sizes(); const auto nrows = input_sizes[0]; const auto ncols = input_sizes[1]; - auto output = - at::empty({nrows, ncols}, input.options().dtype(at::kHalf)); + auto output = at::empty({nrows, ncols}, input.options().dtype(at::kHalf)); FloatToBFloat16Quantized_ref( input.const_data_ptr(), @@ -71,8 +70,8 @@ at::Tensor _bfloat16_to_float_cpu(const at::Tensor& input) { const auto ncols = input_sizes[1]; auto output = at::empty( - {nrows, ncols}, // 4 = sizeof(float) - input.options().dtype(at::kFloat)); // + {nrows, ncols}, + input.options().dtype(at::kFloat)); BFloat16QuantizedToFloat_ref( reinterpret_cast(input.const_data_ptr()), nrows,