llvm
/
torch-mlir
mirror of https://github.com/llvm/torch-mlir
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Adapt to ByteCodeOpInterface changes.

integrate_iree_20230911
Stella Laurenzo 2023-09-11 11:13:48 -07:00
parent 47452cc77b
commit 86b16ee84c
2 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
include/torch-mlir/Dialect/Torch/IR/TorchOps.h
View File

@ -10,6 +10,7 @@
#ifndef TORCHMLIR_DIALECT_TORCH_IR_TORCHOPS_H
#define TORCHMLIR_DIALECT_TORCH_IR_TORCHOPS_H
#include "mlir/Bytecode/BytecodeOpInterface.h"
#include "mlir/IR/BuiltinTypes.h"
#include "mlir/IR/Matchers.h"
#include "mlir/IR/OpDefinition.h"

3
lib/Dialect/Torch/IR/CMakeLists.txt
View File

@ -16,6 +16,9 @@ add_mlir_library(TorchMLIRTorchDialect
Core
LINK_LIBS PUBLIC
MLIRBytecodeOpInterface
MLIRBytecodeReader
MLIRBytecodeWriter
MLIRFuncDialect
MLIRIR
MLIRSupport