Skip to content

Commit 5dd64ea

Browse files
[pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
1 parent 3005dbe commit 5dd64ea

2 files changed

Lines changed: 2 additions & 2 deletions

File tree

transformer_engine/pytorch/tensor/mxfp8_tensor.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -394,7 +394,7 @@ def untyped_storage(self) -> torch.UntypedStorage:
394394
data = self._rowwise_data if self._rowwise_data is not None else self._columnwise_data
395395
if data is not None:
396396
return data.untyped_storage()
397-
return self._default_storage # Unique 1-byte storage.
397+
return self._default_storage # Unique 1-byte storage.
398398

399399
@classmethod
400400
def __torch_dispatch__(cls, func, types, args, kwargs=None):

transformer_engine/pytorch/tensor/nvfp4_tensor.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -561,7 +561,7 @@ def untyped_storage(self) -> torch.UntypedStorage:
561561
data = self._rowwise_data if self._rowwise_data is not None else self._columnwise_data
562562
if data is not None:
563563
return data.untyped_storage()
564-
return self._default_storage # Unique 1-byte storage.
564+
return self._default_storage # Unique 1-byte storage.
565565

566566
@classmethod
567567
def __torch_dispatch__(cls, func, types, args, kwargs=None):

0 commit comments

Comments
 (0)