File tree Expand file tree Collapse file tree 1 file changed +0
-12
lines changed
transformer_engine/pytorch/csrc Expand file tree Collapse file tree 1 file changed +0
-12
lines changed Original file line number Diff line number Diff line change @@ -541,18 +541,6 @@ Float8BlockQuantizer::Float8BlockQuantizer(const py::handle& quantizer) : Quanti
541541}
542542
543543void 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
558546std::pair<TensorWrapper, py::object> Float8BlockQuantizer::create_tensor (
You can’t perform that action at this time.
0 commit comments