//===- MLProgramAttributed.td - Attr definitions -----------*- tablegen -*-===//
//
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
// See https://llvm.org/LICENSE.txt for license information.
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
//
//===----------------------------------------------------------------------===//

#ifndef MLPROGRAM_ATTRIBUTES
#define MLPROGRAM_ATTRIBUTES

include "mlir/IR/AttrTypeBase.td"
include "mlir/IR/BuiltinAttributeInterfaces.td"
include "mlir/Dialect/MLProgram/IR/MLProgramBase.td"

// Base class for MLProgram dialect attributes.
class MLProgram_Attr<string name, list<Trait> traits = []>
    : AttrDef<MLProgram_Dialect, name, traits> {
  let mnemonic = ?;
}

//===----------------------------------------------------------------------===//
// ExternAttr
//===----------------------------------------------------------------------===//

def MLProgram_ExternAttr : MLProgram_Attr<"Extern", [TypedAttrInterface]> {
  let summary = "Value used for a global signalling external resolution";
  let description = [{
  When used as the value for a GlobalOp, this indicates that the actual
  value should be resolved externally in an implementation defined manner.
  The `sym_name` of the global is the key for locating the value.

  Examples:

  ```mlir
  extern : tensor<4xi32>
  ```
  }];

  let parameters = (ins AttributeSelfTypeParameter<"">:$type);
  let mnemonic = "extern";
  let assemblyFormat = "";

  // Generate mnemonic alias for the attribute.
  let genMnemonicAlias = 1;
}

#endif // MLPROGRAM_ATTRIBUTES