Skip to content

Commit 4cd6a67

Browse files
[pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
1 parent b53cc47 commit 4cd6a67

File tree

1 file changed

+1
-1
lines changed
  • transformer_engine/pytorch/csrc/extensions

1 file changed

+1
-1
lines changed

transformer_engine/pytorch/csrc/extensions/cast.cpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1099,7 +1099,7 @@ std::vector<py::object> split_quantize(const at::Tensor &tensor,
10991099
uint8_t *input_dptr = reinterpret_cast<uint8_t *>(input_py.data_ptr());
11001100
auto input_dtype = GetTransformerEngineDType(input_py.scalar_type());
11011101
NVTEShape input_shape;
1102-
input_shape.ndim=0;
1102+
input_shape.ndim = 0;
11031103
size_t input_size = 1;
11041104
for (const auto &d : input_py.sizes()) {
11051105
input_shape.data[input_shape.ndim++] = static_cast<size_t>(d);

0 commit comments

Comments
 (0)