add PIM accelerator

src/PIM/PimAccelerator.hpp

@@ -0,0 +1,70 @@
+#pragma once
+
+#include "mlir/IR/BuiltinTypes.h"
+#include "src/Accelerators/Accelerator.hpp"
+
+namespace onnx_mlir {
+namespace accel {
+
+/// Singleton class to construct PIM accelerator.
+class PimAccelerator final : public Accelerator {
+private:
+  static PimAccelerator *instance;
+  PimAccelerator();
+
+public:
+  /// Singleton should not be clonable or assignable.
+  PimAccelerator(PimAccelerator &) = delete;
+  void operator=(const PimAccelerator &) = delete;
+
+  ~PimAccelerator();
+
+  /// Creates an instance on the first invocation. Subsequent invocations
+  /// return the existing instance.
+  static PimAccelerator *getInstance();
+
+  /// Define classof to be able to use isa<>, cast<>, dyn_cast<>, etc.
+  static bool classof(const Accelerator *accel) {
+    return accel->getKind() == Accelerator::Kind::PIM;
+  }
+  static bool classof(const PimAccelerator *) { return true; }
+
+  uint64_t getVersionNumber() const final;
+
+  //===--------------------------------------------------------------------===//
+  // Hooks for onnx-mlir-opt driver
+  //===--------------------------------------------------------------------===//
+  virtual void addPasses(mlir::OwningOpRef<mlir::ModuleOp> &module,
+      mlir::PassManager &pm, onnx_mlir::EmissionTargetType &emissionTarget,
+      std::string outputNameNoExt) const final;
+  //===--------------------------------------------------------------------===//
+  // Hooks for onnx-mlir-opt driver
+  //===--------------------------------------------------------------------===//
+  virtual void registerDialects(mlir::DialectRegistry &registry) const final;
+  virtual void registerPasses(int optLevel) const final;
+  //===--------------------------------------------------------------------===//
+  // Hooks for both onnx-mlir and onnx-mlir-opt drivers
+  //===--------------------------------------------------------------------===//
+  virtual void configurePasses() const final;
+  //===--------------------------------------------------------------------===//
+  // Hooks for onnx-to-krnl pass
+  //===--------------------------------------------------------------------===//
+  virtual mlir::MemRefType convertTensorTypeToMemRefType(
+      const mlir::TensorType tensorType) const final;
+  virtual void conversionTargetONNXToKrnl(
+      mlir::ConversionTarget &target) const final;
+  virtual void rewritePatternONNXToKrnl(mlir::RewritePatternSet &patterns,
+      mlir::TypeConverter &typeConverter, mlir::MLIRContext *ctx) const final;
+
+  //===--------------------------------------------------------------------===//
+  // Hooks for krnl-to-llvm pass
+  //===--------------------------------------------------------------------===//
+  virtual void conversionTargetKrnlToLLVM(
+      mlir::ConversionTarget &target) const final;
+  virtual void rewritePatternKrnlToLLVM(mlir::RewritePatternSet &patterns,
+      mlir::LLVMTypeConverter &typeConverter,
+      mlir::MLIRContext *ctx) const final;
+};
+
+} // namespace accel
+} // namespace onnx_mlir