diff options
Diffstat (limited to 'mlir/lib/Dialect/NVGPU/Transforms/MmaSyncTF32Transform.cpp')
| -rw-r--r-- | mlir/lib/Dialect/NVGPU/Transforms/MmaSyncTF32Transform.cpp | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/mlir/lib/Dialect/NVGPU/Transforms/MmaSyncTF32Transform.cpp b/mlir/lib/Dialect/NVGPU/Transforms/MmaSyncTF32Transform.cpp index 292738de4b52..b9dd174a6b25 100644 --- a/mlir/lib/Dialect/NVGPU/Transforms/MmaSyncTF32Transform.cpp +++ b/mlir/lib/Dialect/NVGPU/Transforms/MmaSyncTF32Transform.cpp @@ -42,7 +42,7 @@ struct MmaSyncF32ToTF32Pattern : public OpRewritePattern<nvgpu::MmaSyncOp> { Location location = op->getLoc(); if (op->hasAttr(op.getTf32EnabledAttrName()) || - !op.getMatrixA().getType().cast<VectorType>().getElementType().isF32()) + !cast<VectorType>(op.getMatrixA().getType()).getElementType().isF32()) return failure(); if (precision == MmaSyncF32Lowering::Unkown) |