triton icon indicating copy to clipboard operation
triton copied to clipboard

Implement scaled_dot(mxfp8, fp8) via mma

Open lezcano opened this issue 1 year ago • 2 comments

Initial implementation using mma.

Missing to test that it plays ball with the pipeliner.

lezcano avatar Sep 24 '24 17:09 lezcano

So this breaks the pipeliner somehow?

No, it's just that I haven't tested if it works as expected just yet.

lezcano avatar Sep 30 '24 13:09 lezcano

BTW, it would be nice to have lit tests for accelerate_matmul as well lowering to of upscale to llvm

ThomasRaoux avatar Sep 30 '24 13:09 ThomasRaoux

@ThomasRaoux addressed the review (at long last)

lezcano avatar Oct 11 '24 14:10 lezcano