formatting

This commit is contained in:
cm2435 2024-02-04 12:47:49 +00:00
commit e3e41a7cef

View file

@ -63,6 +63,7 @@ def _seeded_dropout(
tl.store(output_ptr + off2, o2, mask = mask2)
tl.store(output_ptr + off3, o3, mask = mask3)
pass
def seeded_dropout(
x: torch.Tensor,