From fd793093b1075539e0e7b195c92005f39747b147 Mon Sep 17 00:00:00 2001 From: Yi Wang Date: Sat, 6 May 2023 16:12:34 -0700 Subject: [PATCH] Update Triton kernel example --- docs/index.md | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/docs/index.md b/docs/index.md index 5d45376..b2c9a72 100644 --- a/docs/index.md +++ b/docs/index.md @@ -45,7 +45,7 @@ def add_kernel( pid = tl.program_id(axis=0) block_start = pid * block_size offsets = block_start + tl.arange(0, block_size) - mask = offsets < 8 + mask = offsets < block_size x = tl.load(x_ptr + offsets, mask=mask) y = tl.load(y_ptr + offsets, mask=mask) output = x + y