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

Fixes some typos in utilities.md

Browse Source
This commit is contained in:
Zheng Zeng 2021-05-06 22:37:37 +08:00 committed by GitHub
parent 8f8a80cad5
commit b878c96421
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); minimum);
// Initialize in device memory // Initialize in device memory
cutlass::reference::device::TensorFill( cutlass::reference::device::TensorFillRandomUniform(
tensor.device_view(), tensor.device_view(),
seed, seed,
maximum, 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. The device-side implementation uses CURAND to generate random numbers.
```c++ ```c++
#include <cutlass/layout/matrix.h> #include <cutlass/layout/matrix.h>