Skip to content

Commit 29ff982

Browse files
dearmachilikian24
andauthored
Add grid definition to Quick Start (#116)
Co-authored-by: Sudhakar Mahalingam <76498452+likian24@users.noreply.github.com>
1 parent 44c4823 commit 29ff982

1 file changed

Lines changed: 5 additions & 1 deletion

File tree

README.md

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -40,10 +40,14 @@ end
4040
# Launch
4141
vector_size = 2^20
4242
tile_size = 16
43+
44+
blocks = cld(vector_size, tile_size)
45+
grid = (blocks, 1, 1)
46+
4347
a, b = CUDA.rand(Float32, vector_size), CUDA.rand(Float32, vector_size)
4448
c = CUDA.zeros(Float32, vector_size)
4549

46-
ct.launch(vadd, (cld(vector_size, tile_size), 1, 1), a, b, c, ct.Constant(tile_size))
50+
ct.launch(vadd, grid, a, b, c, ct.Constant(tile_size))
4751

4852
@assert c == a .+ b
4953
```

0 commit comments

Comments
 (0)