add_mlir_dialect_library(NPCOMPTorchDialect
  TorchDialect.cpp
  TorchOps.cpp

  ADDITIONAL_HEADER_DIRS
  ${PROJECT_SOURCE_DIR}/include/npcomp/Dialect/Torch

  DEPENDS
  MLIRTorchOpsIncGen

  LINK_COMPONENTS
  Core

  LINK_LIBS PUBLIC
  MLIRIR
  MLIRSupport
  NPCOMPBasicpyDialect
  NPCOMPNumpyDialect
)