From c0daffc625a6ec4d6a8222e6858cdd3096e27ff9 Mon Sep 17 00:00:00 2001 From: Rohit Dwivedula Date: Thu, 2 Sep 2021 07:20:17 +0530 Subject: [PATCH] [DOCS] @heuristics -> @triton.heuristics in some snippets (#253) --- python/triton/code_gen.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python/triton/code_gen.py b/python/triton/code_gen.py index 8b765a5c8..6a301fd1a 100644 --- 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