Skip to content

Commit 9be2661

Browse files
authored
[AMD][Tutorials] Set NUM_GPRS to NUM_REGS by default (#6935)
Set `NUM_GPRS` to `NUM_REGS` by default Signed-off-by: Ilya Veselov <[email protected]>
1 parent fb93fc1 commit 9be2661

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

python/tutorials/02-fused-softmax.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -153,6 +153,7 @@ def softmax(x):
153153
# with matrix VALU instructions, and can also be loaded directly from memory. A wave may have up to 512 total
154154
# VGPRs, 256 of each type. When a wave has fewer than 512 total VGPRs, the number of each type is flexible - it is
155155
# not required to be equal numbers of both types.
156+
NUM_GPRS = NUM_REGS
156157
if is_cdna():
157158
NUM_GPRS = NUM_REGS * 2
158159

0 commit comments

Comments
 (0)