search

srush / Triton-Puzzles

Puzzles for learning Triton
Apache License 2.0
925 stars 57 forks source link

Puzzle 6 - shape annotation for x #11

Closed lucaslingle closed 4 months ago

lucaslingle commented 4 months ago

For "Puzzle 6: Fused Outer Multiplication - Backwards", x appears to have the same meaning as in the previous problem. However, it is annotated as having shape [90, 100] instead of [100]. Is this a typo?

srush commented 4 months ago

Ah, sorry there is a bug in the math there. the Shape 90, 100 is correct, but it should be x_{j, i} in the math.

lucaslingle commented 4 months ago

I figured out, thanks!