[FRONTEND][BACKEND] Small fixes to multiple_of, num_programs, axisinfo; enable block-sparse tests (#927)

2022-11-29 20:00:34 +01:00
parent 66c36c4378
commit 9bb54402b3
8 changed files with 229 additions and 17 deletions
--- a/lib/Conversion/TritonGPUToLLVM/TritonGPUToLLVM.cpp
+++ b/lib/Conversion/TritonGPUToLLVM/TritonGPUToLLVM.cpp
@@ -2074,7 +2074,7 @@ struct GetNumProgramsOpConversion
    Location loc = op->getLoc();
    assert(op.axis() < 3);

-    Value blockId = rewriter.create<::mlir::gpu::BlockDimOp>(
+    Value blockId = rewriter.create<::mlir::gpu::GridDimOp>(
        loc, rewriter.getIndexType(), dims[op.axis()]);
    auto llvmIndexTy = getTypeConverter()->getIndexType();
    rewriter.replaceOpWithNewOp<UnrealizedConversionCastOp>(