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

Reference lazy graph executor (#1507) 

* Add LazyGraphExecutor registration

* Update PyTorch version to 1.14.0.dev20221024

Co-authored-by: Roll PyTorch Action <torch-mlir@users.noreply.github.com>
pull/1524/head snapshot-20221025.637
Jae Hoon (Antonio) Kim 2022-10-24 17:15:11 -04:00 committed by GitHub
parent 3f883802e8
commit 2f300935bf
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
3 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
python/torch_mlir/csrc/reference_lazy_backend/backend_impl.cpp
View File

@ -11,6 +11,7 @@
#include <torch/csrc/lazy/backend/backend_data.h>
#include <torch/csrc/lazy/backend/backend_device.h>
#include <torch/csrc/lazy/backend/lowering_context.h>
#include <torch/csrc/lazy/core/lazy_graph_executor.h>
#include <torch/csrc/lazy/core/shape.h>
#include <torch_mlir/csrc/base_lazy_backend/backend_impl.h>
@ -186,6 +187,9 @@ void InitReferenceLazyBackend() {
at::RegisterTorchMlirLazyNativeFunctions();
static std::unique_ptr<BackendRegistrar> g_registrar;
g_registrar.reset(new BackendRegistrar(GetReferenceLazyBackendImpl()));
static LazyGraphExecutor* executor = new LazyGraphExecutor();
LazyGraphExecutor::Register(executor);
}
ComputationPtr& GetLatestComputation() {

2
pytorch-requirements.txt
View File

@ -1,3 +1,3 @@
-f https://download.pytorch.org/whl/nightly/cpu/torch_nightly.html
--pre
torch==1.14.0.dev20221021
torch==1.14.0.dev20221024

2
pytorch-version.txt
View File

@ -1 +1 @@
6418c7d4ba409bc91dd0c2638dc541ae54de75fc
9c14af470f14afd379255e3d8292db680718c4db