gaspersic/triton
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
Rohit Dwivedula
2021-09-02 07:20:17 +05:30
committed by GitHub
parent 274d613488
commit c0daffc625
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
python/triton/code_gen.py
View File

@@ -794,7 +794,7 @@ def heuristics(values):
.. highlight:: python .. highlight:: python
.. code-block:: 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 @triton.jit
def kernel(x_ptr, x_size, **META): def kernel(x_ptr, x_size, **META):
BLOCK_SIZE = META['BLOCK_SIZE'] # smallest power-of-two >= x_size BLOCK_SIZE = META['BLOCK_SIZE'] # smallest power-of-two >= x_size