mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-17 04:16:07 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix example in mosaic tpu dialect layout.h

Browse Source 
PiperOrigin-RevId: 629424833
This commit is contained in:
jax authors 2024-04-30 08:42:05 -07:00
parent 20a8e2a6ec
commit eeca8d81b9
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
jaxlib/mosaic/dialect/tpu/layout.h
View File

@ -139,13 +139,13 @@ class RectangularVregBounds : public VRegDataBounds {
// . . . . . . . . . .
// . . a b c d e . . .
// . . f g h i j . . .
// . . . . . . . . . .
// . . k l m n o . . .
//
// vreg 3 vreg 4
// . . k l m n o . . .
// . . p q r s t . . .
// . . . . . . . . . .
// . . . . . . . . . .
// . . . . . . . . . .
//
// The dot character indicates padding. Nothing should be assumed about the
// value of those entries.