[DOCS] @heuristics -> @triton.heuristics in some snippets (#253)

2021-09-02 07:20:17 +05:30
parent 274d613488
commit c0daffc625
1 changed files with 1 additions and 1 deletions
--- a/python/triton/code_gen.py
+++ b/python/triton/code_gen.py
@@ -794,7 +794,7 @@ def heuristics(values):
    .. highlight:: python
    .. code-block:: python

-        @heuristics(values={'BLOCK_SIZE': lambda args: 2 ** int(math.ceil(math.log2(args[1])))})
+        @triton.heuristics(values={'BLOCK_SIZE': lambda args: 2 ** int(math.ceil(math.log2(args[1])))})
        @triton.jit
        def kernel(x_ptr, x_size, **META):
            BLOCK_SIZE = META['BLOCK_SIZE'] # smallest power-of-two >= x_size