Index: mlir/include/mlir/Dialect/LLVMIR/NVVMOps.td
===================================================================
--- mlir/include/mlir/Dialect/LLVMIR/NVVMOps.td
+++ mlir/include/mlir/Dialect/LLVMIR/NVVMOps.td
@@ -29,6 +29,7 @@
   let cppNamespace = "::mlir::NVVM";
   let dependentDialects = ["LLVM::LLVMDialect"];
   let hasOperationAttrVerify = 1;
+  let useFoldAPI = kEmitFoldAdaptorFolder;
 
   let extraClassDeclaration = [{
     /// Get the name of the attribute used to annotate external kernel
Index: mlir/include/mlir/Dialect/LLVMIR/ROCDLOps.td
===================================================================
--- mlir/include/mlir/Dialect/LLVMIR/ROCDLOps.td
+++ mlir/include/mlir/Dialect/LLVMIR/ROCDLOps.td
@@ -25,6 +25,7 @@
   let cppNamespace = "::mlir::ROCDL";
   let dependentDialects = ["LLVM::LLVMDialect"];
   let hasOperationAttrVerify = 1;
+  let useFoldAPI = kEmitFoldAdaptorFolder;
 
   let extraClassDeclaration = [{
     /// Get the name of the attribute used to annotate external kernel
Index: mlir/test/python/python_test_ops.td
===================================================================
--- mlir/test/python/python_test_ops.td
+++ mlir/test/python/python_test_ops.td
@@ -20,6 +20,7 @@
 
   let useDefaultTypePrinterParser = 1;
   let useDefaultAttributePrinterParser = 1;
+  let useFoldAPI = kEmitFoldAdaptorFolder;
 }
 
 class TestType<string name, string typeMnemonic>