squall/cutlass
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

Merge pull request #256 from zheng95z/patch-2

Browse Source 
Fixes some typos in utilities.md
This commit is contained in:
Haicheng Wu 2021-05-06 11:02:49 -04:00 committed by GitHub
commit 9840d25269
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
media/docs/utilities.md
View File

@ -200,7 +200,7 @@ int main() {
minimum);
// Initialize in device memory
cutlass::reference::device::TensorFill(
cutlass::reference::device::TensorFillRandomUniform(
tensor.device_view(),
seed,
maximum,
@ -211,7 +211,7 @@ int main() {
```
`TensorFillRandomUniform()` for initializing elements to a random uniform distribution.
`TensorFillRandomGaussian()` for initializing elements to a random Gaussian distribution.
The device-side implementation uses CURAND to generate random numbers.
```c++
#include <cutlass/layout/matrix.h>