Skip to content

Commit 9f06212

Browse files
committed
missed a quant code removal
1 parent b041111 commit 9f06212

File tree

1 file changed

+0
-12
lines changed

1 file changed

+0
-12
lines changed

transformer_engine/pytorch/csrc/quantizer.cpp

Lines changed: 0 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -541,18 +541,6 @@ Float8BlockQuantizer::Float8BlockQuantizer(const py::handle& quantizer) : Quanti
541541
}
542542

543543
void Float8BlockQuantizer::set_quantization_params(TensorWrapper* tensor) const {
544-
// Change the rowwise and columnwise_data to the configured dtype.
545-
// May be a switch between E5M2 and E4M3.
546-
auto rowwise_data = tensor->get_rowwise_data();
547-
rowwise_data.dtype = static_cast<NVTEDType>(dtype);
548-
549-
auto columnwise_data = tensor->get_columnwise_data();
550-
columnwise_data.dtype = static_cast<NVTEDType>(dtype);
551-
552-
tensor->set_rowwise_data(rowwise_data.data_ptr, static_cast<DType>(rowwise_data.dtype),
553-
rowwise_data.shape);
554-
tensor->set_columnwise_data(columnwise_data.data_ptr, static_cast<DType>(columnwise_data.dtype),
555-
columnwise_data.shape);
556544
}
557545

558546
std::pair<TensorWrapper, py::object> Float8BlockQuantizer::create_tensor(

0 commit comments

Comments
 (0)