diff --git a/enzyme/BUILD b/enzyme/BUILD index d15ae7a1bad..2f33ca8ef2f 100644 --- a/enzyme/BUILD +++ b/enzyme/BUILD @@ -649,6 +649,7 @@ cc_binary( "@llvm-project//mlir:OpenMPDialect", "@llvm-project//mlir:Pass", "@llvm-project//mlir:SCFDialect", + "@llvm-project//mlir:TensorDialect", "@llvm-project//mlir:Transforms", ], )