Skip to content

Commit 9de60b4

Browse files
zou3519svekars
andauthored
Update recipes_source/torch_compile_user_defined_triton_kernel_tutorial.py
Co-authored-by: Svetlana Karslioglu <[email protected]>
1 parent e5aaa9a commit 9de60b4

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

recipes_source/torch_compile_user_defined_triton_kernel_tutorial.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -337,7 +337,10 @@ def _(x_shape):
337337

338338
x = torch.randn(3, device="cuda")
339339

340-
# NB: FlopCounterMode requires tabulate.
340+
#########################################################
341+
# ``FlopCounterMode`` requires `tabulate <https://pypi.org/project/tabulate/>`__.
342+
# Before running the code below, make sure you have ``tabulate`` installed or install by
343+
# running ``pip install tabulate``.
341344
#
342345
# >>> with FlopCounterMode() as flop_counter:
343346
# >>> y = mysin(x)

0 commit comments

Comments
 (0)