import * as fs from 'fs/promises'; import * as path from 'path'; import * as tablegen from './tablegen.js'; import * as url from 'url'; class Operator { constructor(def) { this.def = def; let opInfo = null; for (const parent of this.def.parents) { const parentClass = this.def.parser.classes.get(parent.name); if (parentClass) { opInfo = this._findOpParent(parentClass, parent.args, {}); if (opInfo) { break; } } } this.dialectName = opInfo?.dialect || null; this.opName = opInfo?.mnemonic || null; } getDialectName() { return this.dialectName || ''; } getOperationName() { return this.dialectName && this.opName ? `${this.dialectName}.${this.opName}` : null; } _extractValue(arg) { // Handle both old string format and new Value object format if (typeof arg === 'string') { return arg; } if (arg && typeof arg === 'object') { if (arg.value !== undefined && arg.name !== undefined) { return this._extractValue(arg.value); } if (arg.type === 'string' && typeof arg.value === 'string') { return arg.value.replace(/^"|"$/g, ''); } if (arg.type === 'def' && typeof arg.value === 'string') { return arg.value; } } return null; } _evaluateWithSubstitutions(value, subs, visited = new Set()) { if (!value) { return null; } // Detect cycles const valueKey = JSON.stringify(value); if (visited.has(valueKey)) { return null; } visited.add(valueKey); switch (value.type) { case 'string': return typeof value.value === 'string' ? value.value.replace(/^"|"$/g, '') : value.value; case 'int': return parseInt(value.value, 10); case 'concat': { const parts = value.value.map((v) => this._evaluateWithSubstitutions(v, subs, visited)); const filtered = parts.filter((v) => v !== null && v !== undefined && v !== ''); const result = filtered.join(''); return result; } case 'id': case 'def': { const name = value.value; if (subs[name] && subs[name] !== value) { return this._evaluateWithSubstitutions(subs[name], subs, visited); } return this.def.evaluateValue(value); } default: { return this.def.evaluateValue(value); } } } _findOpParent(parentClass, parentArgs, substitutions) { const subs = { ...substitutions }; if (parentClass.templateArgs && parentArgs) { for (let i = 0; i < Math.min(parentClass.templateArgs.length, parentArgs.length); i++) { const paramName = parentClass.templateArgs[i].name; const argValue = parentArgs[i]; const extractedValue = this._extractValue(argValue); if (extractedValue && substitutions[extractedValue]) { subs[paramName] = substitutions[extractedValue]; } else { const evaluated = this._evaluateWithSubstitutions(argValue, substitutions); subs[paramName] = evaluated === null ? argValue : subs[paramName] = { type: 'string', value: evaluated }; } } } if (parentClass.name === 'Op' && parentArgs.length >= 2) { let [dialectArg, mnemonicArg] = parentArgs; if (dialectArg && dialectArg.type === 'def' && dialectArg.value && subs[dialectArg.value]) { dialectArg = subs[dialectArg.value]; } if (mnemonicArg && mnemonicArg.type === 'def' && mnemonicArg.value && subs[mnemonicArg.value]) { mnemonicArg = subs[mnemonicArg.value]; } let dialectName = null; const dialectStr = this._extractValue(dialectArg); if (dialectStr) { const dialectDef = this.def.parser.getDef(dialectStr) || this.def.parser.getClass(dialectStr); if (dialectDef) { dialectName = dialectDef.getValueAsString('name'); } } let mnemonic = this._extractValue(mnemonicArg); if (!mnemonic && mnemonicArg) { mnemonic = this._evaluateWithSubstitutions(mnemonicArg, subs); } // Clean up mnemonic: remove leading/trailing dots, normalize multiple dots if (mnemonic && typeof mnemonic === 'string') { mnemonic = mnemonic.replace(/^\.+|\.+$/g, ''); // Remove leading/trailing dots // Skip if mnemonic is invalid after cleanup if (!mnemonic || mnemonic.includes('..')) { return null; } } if (dialectName && mnemonic) { return { dialect: dialectName, mnemonic }; } } for (const grandparent of parentClass.parents) { const grandparentClass = this.def.parser.classes.get(grandparent.name); if (grandparentClass) { const resolvedArgs = grandparent.args.map((arg) => { const extracted = this._extractValue(arg); return (extracted && subs[extracted]) ? subs[extracted] : arg; }); const result = this._findOpParent(grandparentClass, resolvedArgs, subs); if (result) { return result; } } } return null; } } const access = async (path) => { try { await fs.access(path); return true; } catch { return false; } }; const main = async () => { const dirname = path.dirname(url.fileURLToPath(import.meta.url)); const source = path.join(dirname, '..', 'third_party', 'source', 'mlir'); const paths = [ path.join(source, 'llvm-project', 'mlir', 'include'), path.join(source, 'llvm-project', 'mlir', 'test', 'lib', 'Dialect', 'Transform'), path.join(source, 'llvm-project', 'mlir', 'include', 'mlir', 'Dialect', 'ArmNeon'), path.join(source, 'llvm-project', 'mlir', 'include', 'mlir', 'Dialect', 'ArmSME', 'IR'), path.join(source, 'llvm-project', 'mlir', 'include', 'mlir', 'Dialect', 'ArmSVE', 'IR'), path.join(source, 'llvm-project', 'mlir', 'examples', 'toy', 'Ch7', 'include'), path.join(source, 'stablehlo'), path.join(source, 'onnx-mlir'), path.join(source, 'torch-mlir', 'include'), path.join(source, 'triton', 'include'), path.join(source, 'triton', 'third_party'), path.join(source, 'triton', 'third_party', 'amd', 'include', 'Dialect', 'TritonAMDGPU', 'IR'), path.join(source, 'mlir-hlo', 'include'), path.join(source, 'iree', 'compiler', 'src'), path.join(source, 'FlashTensor', 'include'), path.join(source, 'tpu-mlir', 'include'), path.join(source, 'tensorflow'), path.join(source, 'tensorflow', 'tensorflow', 'compiler', 'mlir', 'tfrt', 'ir'), path.join(source, 'runtime', 'include'), path.join(source, 'plaidml'), path.join(source, 'plaidml', 'pmlc', 'dialect', 'pxa', 'ir'), path.join(source, 'mlir-dace', 'include'), path.join(source, 'lltz', 'mlir', 'dialect', 'include', 'Michelson'), path.join(source, 'lagrad', 'include', 'LAGrad'), path.join(source, 'TensorRT-Incubator', 'mlir-tensorrt', 'tensorrt', 'include'), path.join(source, 'TensorRT-Incubator', 'mlir-tensorrt', 'executor', 'include'), ]; const dialects = [ 'mlir/IR/BuiltinAttributeInterfaces.td', 'mlir/IR/BuiltinTypeInterfaces.td', 'mlir/IR/BuiltinLocationAttributes.td', 'mlir/IR/BuiltinDialect.td', 'mlir/IR/BuiltinOps.td', 'mlir/IR/BuiltinDialectBytecode.td', 'mlir/IR/BuiltinAttributes.td', 'mlir/IR/BuiltinTypes.td', 'mlir/Dialect/Async/IR/AsyncOps.td', 'mlir/Dialect/Affine/IR/AffineOps.td', 'mlir/Dialect/Affine/IR/AffineOps.td', 'mlir/Dialect/Arith/IR/ArithOps.td', 'mlir/Dialect/ControlFlow/IR/ControlFlowOps.td', 'mlir/Dialect/Func/IR/FuncOps.td', 'mlir/Dialect/GPU/IR/GPUOps.td', 'mlir/Dialect/SCF/IR/SCFOps.td', 'mlir/Dialect/Linalg/IR/LinalgOps.td', // 'mlir/Dialect/Linalg/IR/LinalgStructuredOps.td', // File not found 'mlir/Dialect/Linalg/IR/LinalgNamedStructuredOps.yamlgen.td' 'mlir/Dialect/Linalg/IR/LinalgRelayoutOps.td', 'mlir/Dialect/MemRef/IR/MemRefOps.td', 'mlir/Dialect/Bufferization/IR/BufferizationOps.td', 'mlir/Dialect/Quant/IR/QuantOps.td', 'mlir/Dialect/Shape/IR/ShapeOps.td', 'mlir/Dialect/SparseTensor/IR/SparseTensorOps.td', 'mlir/Dialect/Tensor/IR/TensorOps.td', 'mlir/Dialect/Tosa/IR/TosaOps.td', 'mlir/Dialect/Vector/IR/VectorOps.td', 'mlir/Dialect/X86Vector/X86Vector.td', 'mlir/Dialect/XeGPU/IR/XeGPUOps.td', 'mlir/Dialect/Transform/IR/TransformOps.td', 'mlir/Dialect/Transform/DebugExtension/DebugExtensionOps.td', 'mlir/Dialect/Transform/IRDLExtension/IRDLExtensionOps.td', 'mlir/Dialect/Transform/LoopExtension/LoopExtensionOps.td', 'mlir/Dialect/Transform/PDLExtension/PDLExtensionOps.td', 'mlir/Dialect/Transform/SMTExtension/SMTExtensionOps.td', 'mlir/Dialect/Transform/TuneExtension/TuneExtensionOps.td', 'TestTransformDialectExtension.td', 'iree/compiler/Dialect/Util/TransformOps/UtilTransformOps.td', 'mlir/Dialect/Linalg/TransformOps/LinalgTransformOps.td', 'mlir/Dialect/Linalg/TransformOps/LinalgMatchOps.td', 'mlir/Dialect/SCF/TransformOps/SCFTransformOps.td', 'mlir/Dialect/Bufferization/TransformOps/BufferizationTransformOps.td', 'mlir/Dialect/GPU/TransformOps/GPUTransformOps.td', 'mlir/Dialect/NVGPU/TransformOps/NVGPUTransformOps.td', 'mlir/Dialect/Affine/TransformOps/AffineTransformOps.td', 'mlir/Dialect/SparseTensor/TransformOps/SparseTensorTransformOps.td', 'mlir/Dialect/Tensor/TransformOps/TensorTransformOps.td', 'mlir/Dialect/Vector/TransformOps/VectorTransformOps.td', 'mlir/Dialect/MemRef/TransformOps/MemRefTransformOps.td', 'mlir/Dialect/Func/TransformOps/FuncTransformOps.td', 'mlir/Dialect/ArmNeon/TransformOps/ArmNeonVectorTransformOps.td', 'mlir/Dialect/ArmSVE/TransformOps/ArmSVEVectorTransformOps.td', 'mlir/Dialect/XeGPU/TransformOps/XeGPUTransformOps.td', 'mlir/Dialect/DLTI/TransformOps/DLTITransformOps.td', 'mlir/Dialect/WasmSSA/IR/WasmSSAOps.td', 'mlir/Dialect/IRDL/IR/IRDLOps.td', 'mlir/Dialect/LLVMIR/LLVMOps.td', 'mlir/Dialect/LLVMIR/LLVMIntrinsicOps.td', 'mlir/Dialect/LLVMIR/NVVMOps.td', 'mlir/Dialect/LLVMIR/ROCDLOps.td', // 'mlir/Dialect/OpenMP/OpenMPOps.td', // File not found 'mlir/Dialect/OpenMP/OmpCommon.td' 'mlir/Dialect/ArmSME/IR/ArmSMEOps.td', 'mlir/Dialect/ArmNeon/ArmNeon.td', 'mlir/Dialect/ArmSVE/IR/ArmSVE.td', 'mlir/Dialect/Math/IR/MathOps.td', 'mlir/Dialect/MLProgram/IR/MLProgramOps.td', 'mlir/Dialect/SPIRV/IR/SPIRVStructureOps.td', 'mlir/Dialect/SPIRV/IR/SPIRVControlFlowOps.td', 'mlir/Dialect/SPIRV/IR/SPIRVArithmeticOps.td', 'mlir/Dialect/SPIRV/IR/SPIRVLogicalOps.td', 'mlir/Dialect/SPIRV/IR/SPIRVBitOps.td', 'mlir/Dialect/SPIRV/IR/SPIRVCastOps.td', 'mlir/Dialect/SPIRV/IR/SPIRVCompositeOps.td', 'mlir/Dialect/SPIRV/IR/SPIRVAtomicOps.td', 'mlir/Dialect/SPIRV/IR/SPIRVBarrierOps.td', 'mlir/Dialect/SPIRV/IR/SPIRVGroupOps.td', 'mlir/Dialect/SPIRV/IR/SPIRVMemoryOps.td', 'mlir/Dialect/SPIRV/IR/SPIRVMiscOps.td', 'mlir/Dialect/SPIRV/IR/SPIRVMatrixOps.td', 'mlir/Dialect/SPIRV/IR/SPIRVImageOps.td', 'mlir/Dialect/SPIRV/IR/SPIRVGLOps.td', 'mlir/Dialect/SPIRV/IR/SPIRVCLOps.td', 'mlir/Dialect/SPIRV/IR/SPIRVNonUniformOps.td', 'mlir/Dialect/SPIRV/IR/SPIRVCooperativeMatrixOps.td', 'mlir/Dialect/SPIRV/IR/SPIRVIntegerDotProductOps.td', 'mlir/Dialect/SPIRV/IR/SPIRVIntelExtOps.td', 'mlir/Dialect/SPIRV/IR/SPIRVGraphOps.td', 'mlir/Dialect/SPIRV/IR/SPIRVMeshOps.td', 'mlir/Dialect/SPIRV/IR/SPIRVPrimitiveOps.td', 'mlir/Dialect/EmitC/IR/EmitC.td', 'mlir/Dialect/Complex/IR/ComplexOps.td', 'mlir/Dialect/Index/IR/IndexOps.td', 'mlir/Dialect/PDL/IR/PDLOps.td', 'mlir/Dialect/Ptr/IR/PtrOps.td', 'mlir/Dialect/UB/IR/UBOps.td', 'mlir/Dialect/AMDGPU/IR/AMDGPU.td', 'mlir/Dialect/NVGPU/IR/NVGPUOps.td', 'mlir/Dialect/Shard/IR/ShardOps.td', 'mlir/Dialect/AMX/AMX.td', 'mlir/Dialect/SMT/IR/SMTOps.td', 'mlir/Dialect/SMT/IR/SMTArrayOps.td', 'mlir/Dialect/SMT/IR/SMTBitVectorOps.td', 'mlir/Dialect/SMT/IR/SMTIntOps.td', // 'mlir/Dialect/OpenACC/OpenACCOps.td', // File not found 'mlir/Dialect/OpenACC/AccCommon.td' (requires full LLVM tree to generate) 'mlir/Dialect/LLVMIR/XeVMOps.td', 'toy/Ops.td', 'stablehlo/dialect/StablehloOps.td', 'stablehlo/dialect/ChloOps.td', 'stablehlo/dialect/VhloOps.td', 'stablehlo/reference/InterpreterOps.td', 'stablehlo/tests/CheckOps.td', 'src/Dialect/ONNX/ONNX.td', 'src/Dialect/ONNX/ONNXOps.td.inc', 'src/Dialect/ONNX/AdditionalONNXOps.td', 'src/Dialect/Krnl/Krnl.td', 'torch-mlir/Dialect/Torch/IR/TorchOps.td', 'torch-mlir/Dialect/TorchConversion/IR/TorchConversionOps.td', 'torch-mlir-dialects/Dialect/TMTensor/IR/TMTensorOps.td', 'tensorflow/compiler/mlir/lite/ir/tfl_ops.td', 'tensorflow/compiler/mlir/tensorflow/ir/tf_ops.td', 'tensorflow/compiler/mlir/tensorflow/ir/tf_saved_model_ops.td', 'tensorflow/compiler/mlir/tensorflow/ir/tf_device_ops.td', 'tensorflow/compiler/mlir/tensorflow/ir/tf_executor_ops.td', 'tensorflow/compiler/mlir/tools/kernel_gen/ir/tf_framework_ops.td', 'tensorflow/compiler/mlir/tfr/ir/tfr_ops.td', 'tensorflow/compiler/mlir/tfrt/ir/tfrt_fallback.td', 'tensorflow/compiler/mlir/tfrt/ir/tfrt_fallback_async.td', 'tensorflow/compiler/mlir/tfrt/ir/tfrt_fallback_sync.td', 'tensorflow/compiler/mlir/tensorflow/ir/host_runtime/tfrt_ops.td', 'tensorflow/compiler/mlir/tfrt/runtime_fallback/runtime_fallback_ops.td', 'tensorflow/compiler/mlir/tfrt/ir/mlrt/mlrt_ops.td', 'tfrt/core_runtime/opdefs/core_runtime.td', 'tfrt/basic_kernels/opdefs/basic_kernels.td', 'tfrt/test_kernels/opdefs/test_kernels.td', 'tfrt/tensor/opdefs/tensor.td', 'tfrt/tensor/opdefs/dense_host_tensor.td', 'mlir-hlo/Dialect/mhlo/IR/hlo_ops.td', 'iree/compiler/Dialect/HAL/IR/HALOps.td', 'iree/compiler/Dialect/HAL/IR/HALTypes.td', 'iree/compiler/Modules/HAL/Loader/IR/HALLoaderOps.td', 'iree/compiler/Modules/HAL/Inline/IR/HALInlineOps.td', 'iree/compiler/Dialect/Flow/IR/FlowOps.td', 'iree/compiler/Dialect/Stream/IR/StreamOps.td', 'iree/compiler/Codegen/Dialect/VectorExt/IR/VectorExtOps.td', 'iree/compiler/Dialect/LinalgExt/IR/LinalgExtOps.td', 'iree/compiler/Dialect/LinalgExt/IR/LinalgExtPureOps.td', 'iree/compiler/Dialect/TensorExt/IR/TensorExtOps.td', 'iree/compiler/Dialect/Util/IR/UtilOps.td', 'iree/compiler/Dialect/VM/IR/VMOps.td', 'iree/compiler/Dialect/VMVX/IR/VMVXOps.td', 'iree/compiler/Codegen/Dialect/GPU/IR/IREEGPUOps.td', 'iree/compiler/Codegen/Dialect/Codegen/IR/IREECodegenOps.td', 'iree/compiler/Dialect/Encoding/IR/EncodingOps.td', 'asuka/Dialect/Asuka/IR/AsukaOps.td', 'tpu_mlir/Dialect/Top/IR/TopOps.td', 'tpu_mlir/Dialect/Tpu/IR/TpuOps.td', 'pmlc/dialect/tile/ir/ops.td', 'pmlc/dialect/stdx/ir/ops.td', // 'pmlc/dialect/pxa/ir/ops.td', // File not found 'mlir/Dialect/Arithmetic/IR/ArithmeticBase.td' 'SDFG/Dialect/Ops.td', 'MichelsonOps.td', 'triton/Dialect/Triton/IR/TritonOps.td', 'triton/Dialect/TritonGPU/IR/TritonGPUOps.td', 'triton/Dialect/Gluon/IR/GluonOps.td', 'triton/Dialect/TritonNvidiaGPU/IR/TritonNvidiaGPUOps.td', 'amd/include/Dialect/TritonAMDGPU/IR/TritonAMDGPUOps.td', 'proton/Dialect/include/Dialect/Proton/IR/ProtonOps.td', 'LAGradOps.td', 'mlir-tensorrt-dialect/TensorRT/IR/TensorRTOps.td', 'mlir-executor/Executor/IR/ExecutorOps.td', ]; const file = path.join(dirname, '..', 'source', 'mlir-metadata.json'); const operations = new Map(); const exists = await access(file); if (exists) { const content = await fs.readFile(file, 'utf-8'); const json = JSON.parse(content); for (const op of json) { if (op.name.endsWith('.') || op.name.includes('..') || op.name.includes('#')) { throw new Error(`Invalid operation name '${op.name}'.`); } if (op.name && !op.name.endsWith('.')) { operations.set(op.name, op); } } } const parser = new tablegen.Reader(); await parser.parse(dialects, paths); for (const def of parser.defs) { const op = new Operator(def); const operationName = op.getOperationName(); if (!operationName) { continue; } if (operationName.endsWith('.') || operationName.includes('..') || operationName.includes('#')) { throw new Error(`Invalid operation name '${operationName}'.`); } const operation = { name: operationName }; let args = def.resolveField('arguments'); // If the field value needs evaluation (e.g., it's a computed field), evaluate it if (args && args.value && (args.value.type === 'id' || args.value.type === 'bang')) { const evaluated = def.evaluateValue(args.value); if (evaluated && typeof evaluated === 'object' && evaluated.operator) { // The evaluation returned a DAG directly args = { value: new tablegen.Value('dag', evaluated) }; } } if (!args || !args.value || args.value.type !== 'dag' || (args.value.value && args.value.value.operands && args.value.value.operands.length === 0)) { for (const parent of def.parents) { if (parent.name === 'Arguments' && parent.args && parent.args.length > 0) { const [dagValue] = parent.args; if (dagValue && dagValue.type === 'dag') { args = { value: dagValue }; } break; } } } const name = operation.name.replace(/^(asuka|stablehlo|chlo|affine|linalg|memref|quant|vector|tosa|tfl|tf|onnx|torch\.aten|gpu)\./, ''); if (['reshape', 'broadcast_in_dim', 'dynamic_reshape', 'Reshape', 'Shape', 'Size', 'ConstantOfShape'].indexOf(name) !== -1) { operation.category = 'Shape'; } else if (['transpose', 'reverse', 'pad', 'Transpose', 'Pad'].indexOf(name) !== -1) { operation.category = 'Transform'; } else if (['slice', 'split', 'dynamic_slice', 'gather', 'scatter', 'Slice', 'Gather', 'Scatter', 'concatenate'].indexOf(name) !== -1) { operation.category = 'Tensor'; } else if (['tanh', 'Sigmoid', 'Tanh', 'Relu', 'Softmax', 'softmax', 'sigmoid', 'relu'].indexOf(name) !== -1) { operation.category = 'Activation'; } else if (['convolution', 'Conv', 'matmul', 'batch_matmul', 'conv2d', 'conv3d', 'fully_connected', 'conv_2d'].indexOf(name) !== -1) { operation.category = 'Layer'; } else if (['batch_norm_inference'].includes(name)) { operation.category = 'Normalization'; } const summary = def.resolveField('summary'); if (summary && summary.value) { const value = def.evaluateValue(summary.value); if (value) { let summary = value.trim(); summary = summary.replace(/^"|"$/g, ''); if (summary) { operation.summary = summary; } } } const description = def.resolveField('description'); if (description && description.value) { const value = def.evaluateValue(description.value); if (value) { let desc = value.trim(); desc = desc.replace(/^\[\{\s*|\s*\}\]$/g, ''); desc = desc.trim(); if (desc) { operation.description = desc; } } } const attributes = []; const inputs = []; const outputs = []; if (args && args.value && args.value.type === 'dag') { const dag = args.value.value; if (dag.operator === 'ins') { for (const operand of dag.operands) { if (!operand.value || !operand.name) { continue; } let typeName = ''; if (operand.value.type === 'def') { typeName = operand.value.value; } else { // Try to extract from other value types typeName = String(operand.value.value); } if (typeName.includes('Attr')) { attributes.push({ name: operand.name, type: typeName }); } else { inputs.push({ name: operand.name, type: typeName }); } } } } let results = def.resolveField('results'); if (!results || !results.value || results.value.type !== 'dag' || (results.value.value && results.value.value.operands && results.value.value.operands.length === 0)) { for (const parent of def.parents) { if (parent.name === 'Results' && parent.args && parent.args.length > 0) { const [dagValue] = parent.args; if (dagValue && dagValue.type === 'dag') { results = { value: dagValue }; } break; } } } if (results && results.value && results.value.type === 'dag') { const dag = results.value.value; if (dag.operator === 'outs') { for (const operand of dag.operands) { if (!operand.value || !operand.name) { continue; } if (operand.value.type !== 'def') { throw new Error('Unexpected result operand value type'); } const type = operand.value.value; outputs.push({ name: operand.name, type }); } } } if (inputs.length > 0) { operation.inputs = inputs; } if (outputs.length > 0) { operation.outputs = outputs; } if (attributes.length > 0) { operation.attributes = attributes; } const successors = def.resolveField('successors'); if (successors && successors.value && successors.value.type === 'dag') { const dag = successors.value.value; if (dag.operator === 'successor') { const successors = []; for (const operand of dag.operands) { if (operand.name) { successors.push({ name: operand.name }); } } if (successors.length > 0) { operation.successors = successors; } } } const assemblyFormat = def.resolveField('assemblyFormat'); if (assemblyFormat && assemblyFormat.value) { const value = def.evaluateValue(assemblyFormat.value); if (value) { const format = value.trim().replace(/^\[\{\s*|\s*\}\]$/g, ''); if (format) { operation.assemblyFormat = format; } } } const hasCustomAssemblyFormat = def.resolveField('hasCustomAssemblyFormat'); if (hasCustomAssemblyFormat && hasCustomAssemblyFormat.value) { operation.hasCustomAssemblyFormat = def.evaluateValue(hasCustomAssemblyFormat.value); } const parser = def.resolveField('parser'); if (parser && parser.value) { operation.parser = 1; } // Extract defaultDialect from OpAsmOpInterface for (const parent of def.parents) { const possibleTraitArgs = parent.args && parent.args.length >= 2 ? [parent.args[1], parent.args[2]] : []; for (const traitsArg of possibleTraitArgs) { if (traitsArg && traitsArg.type === 'list' && traitsArg.value) { for (const trait of traitsArg.value) { const traitName = trait.type === 'def' ? trait.value : null; const traitDag = trait.type === 'dag' && trait.value?.operator ? trait.value.operator : null; if (traitName === 'OpAsmOpInterface' || traitDag === 'DeclareOpInterfaceMethods') { if (traitDag === 'DeclareOpInterfaceMethods' && trait.value?.operands) { if (trait.value.operands.some((operand) => operand.value && operand.value.type === 'list' && operand.value.value.some((method) => method.type === 'string' && method.value === 'getDefaultDialect'))) { const [dialectName] = operationName.split('.'); operation.defaultDialect = dialectName; break; } } const extraClass = def.resolveField('extraClassDeclaration'); if (extraClass && extraClass.value) { const code = def.evaluateValue(extraClass.value); if (code && typeof code === 'string') { const match = code.match(/getDefaultDialect\(\)\s*\{\s*return\s+"(\w+)"/); if (match) { [, operation.defaultDialect] = match; break; } } } } } if (operation.defaultDialect) { break; } } } if (operation.defaultDialect) { break; } } // Only add operation if it has meaningful data beyond just the name if (Object.keys(operation).length > 1) { operations.set(operationName, operation); } } const sorted = Array.from(operations.values()).sort((a, b) => a.name.localeCompare(b.name)); const output = JSON.stringify(sorted, null, 2); const formatted = output.replace(/\{\s+"name":\s+"([^"]+)",\s+"type":\s+"([^"]+)"\s+\}/g, '{ "name": "$1", "type": "$2" }'); await fs.writeFile(file, formatted, 'utf-8'); }; await main();