We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent b53cc47 commit 4cd6a67Copy full SHA for 4cd6a67
transformer_engine/pytorch/csrc/extensions/cast.cpp
@@ -1099,7 +1099,7 @@ std::vector<py::object> split_quantize(const at::Tensor &tensor,
1099
uint8_t *input_dptr = reinterpret_cast<uint8_t *>(input_py.data_ptr());
1100
auto input_dtype = GetTransformerEngineDType(input_py.scalar_type());
1101
NVTEShape input_shape;
1102
- input_shape.ndim=0;
+ input_shape.ndim = 0;
1103
size_t input_size = 1;
1104
for (const auto &d : input_py.sizes()) {
1105
input_shape.data[input_shape.ndim++] = static_cast<size_t>(d);
0 commit comments