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

[DOCS] fix tutorials for v2.0 (#422)

Browse Source 
- Fix meta-parameter usage on tutorials.
- Install tutorial dependencies on CI.
- Switch from `requirements-test.txt` to `extras_require` for test dependencies, and also use it for tutorial dependencies.
- Make some performance tests deterministic.
This commit is contained in:
Madeleine Thompson
2022-01-07 12:34:38 -08:00
committed by GitHub
parent 8bf551ae7a
commit 9801aa7b56
11 changed files with 54 additions and 32 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
python/tutorials/03-matrix-multiplication.py
View File

@@ -237,9 +237,9 @@ def matmul_kernel(
a_ptrs += BLOCK_SIZE_K * stride_ak
b_ptrs += BLOCK_SIZE_K * stride_bk
# you can fuse arbitrary activation functions here
# while the accumulator is still in FP32 !
if meta['ACTIVATION']:
accumulator = meta['ACTIVATION'](accumulator)
# while the accumulator is still in FP32!
if ACTIVATION:
accumulator = ACTIVATION(accumulator)
c = accumulator.to(tl.float16)
# -----------------------------------------------------------