diff options
| author | Yong He <yonghe@outlook.com> | 2023-09-20 22:28:51 -0700 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2023-09-20 22:28:51 -0700 |
| commit | c31f456bce65c897e15b1a429e38b34fafff44e6 (patch) | |
| tree | dc99184fc346cc67e444ac3dfa9322ae4b26e28c /docs/user-guide | |
| parent | c04f5b4970875d60cdf6ac9ea48dda7add4383f3 (diff) | |
Update a1-02-slangpy.md (#3223)
Diffstat (limited to 'docs/user-guide')
| -rw-r--r-- | docs/user-guide/a1-02-slangpy.md | 3 |
1 files changed, 1 insertions, 2 deletions
diff --git a/docs/user-guide/a1-02-slangpy.md b/docs/user-guide/a1-02-slangpy.md index 883ac249c..04021dd94 100644 --- a/docs/user-guide/a1-02-slangpy.md +++ b/docs/user-guide/a1-02-slangpy.md @@ -66,12 +66,11 @@ import slangpy m = slangpy.loadModule('multiply.slang') A = torch.randn((1024,), dtype=torch.float).cuda() -B = torch.randn((1024,), dtype=torch.float).cuda() output = torch.zeros_like(A).cuda() # Number of threads launched = blockSize * gridSize -m.multiply(inputA=A, inputB=B, output=output).launchRaw(blockSize=(32, 1, 1), gridSize=(64, 1, 1)) +m.multiply(input=A, output=output).launchRaw(blockSize=(32, 1, 1), gridSize=(64, 1, 1)) print(output) ``` |