Skip to content

Commit

Permalink
Update gemm_api_3x.md (NVIDIA#1386)
Browse files Browse the repository at this point in the history
Fixed what it seems to be an obvious typo.
  • Loading branch information
RaulPPelaez authored Jul 10, 2024
1 parent 637b159 commit 2448bb5
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion media/docs/gemm_api_3x.md
Original file line number Diff line number Diff line change
Expand Up @@ -357,7 +357,7 @@ struct CollectiveBuilder {
the best performing `CollectiveMma` from the given parameters.

- `ArchTag` is one of the SM architectures tags from `cutlass::arch::Sm*`.
- `OpClass` is one of the operator class tags from `cutlass::arch::Sm*`.
- `OpClass` is one of the operator class tags from `cutlass::arch::OpClass*`.
- `ElementA` and `ElementB` are the logical value types of the A resp. B tensors.
- `ElementAccumulator` is the accumulator type to be used in the instruction.
- `GmemLayoutA` and `GmemLayoutB` are CUTLASS 2.x layout tags, `layout::RowMajor` or `layout::ColumnMajor`.
Expand Down

0 comments on commit 2448bb5

Please sign in to comment.