Skip to content

Commit d176247

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

1 file changed

Lines changed: 1 addition & 0 deletions

File tree

tests/pytorch/test_numerics.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -189,6 +189,7 @@ def get_nvfp4_inp_supported_dtypes(recipe: recipe.Recipe, dtype: torch.dtype) ->
189189
fp8_recipes.append(nvfp4_4over6())
190190
fp8_recipes.append(nvfp4_row_scaled())
191191

192+
192193
def get_causal_attn_mask(sq: int) -> torch.Tensor:
193194
return torch.triu(torch.ones(sq, sq, device="cuda"), diagonal=1).bool()
194195

0 commit comments

Comments
 (0)