frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 21:14:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

[TensorExpr] Rename schedule.{cpp,h} to loopnest.{cpp,h}. (#35119)

Browse Source 
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/35119

Differential Revision: D20567927

Test Plan: Imported from OSS

Pulled By: ZolotukhinM

fbshipit-source-id: 1fb6d03bd4c6e66aca62140d2b537692577f261d
This commit is contained in:
Mikhail Zolotukhin
2020-03-20 23:31:56 -07:00
committed by Facebook GitHub Bot
parent 3342ab89ac
commit 65cea95777
10 changed files with 10 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
caffe2/CMakeLists.txt
View File

@ -485,8 +485,8 @@ if (NOT INTERN_BUILD_MOBILE OR NOT BUILD_CAFFE2_MOBILE)
${TORCH_SRC_DIR}/csrc/jit/tensorexpr/kernel.cpp
${TORCH_SRC_DIR}/csrc/jit/tensorexpr/llvm_codegen.cpp
${TORCH_SRC_DIR}/csrc/jit/tensorexpr/llvm_jit.cpp
${TORCH_SRC_DIR}/csrc/jit/tensorexpr/loopnest.cpp
${TORCH_SRC_DIR}/csrc/jit/tensorexpr/mem_arena.cpp
${TORCH_SRC_DIR}/csrc/jit/tensorexpr/schedule.cpp
${TORCH_SRC_DIR}/csrc/jit/tensorexpr/tensor.cpp
${TORCH_SRC_DIR}/csrc/jit/tensorexpr/types.cpp
${TORCH_SRC_DIR}/csrc/jit/tensorexpr/unique_name_manager.cpp

2
test/cpp/tensorexpr/test_cuda.cpp
View File

@ -9,7 +9,7 @@
#include "test/cpp/tensorexpr/padded_buffer.h"
#include "torch/csrc/jit/tensorexpr/buffer.h"
#include "torch/csrc/jit/tensorexpr/cuda_codegen.h"
#include "torch/csrc/jit/tensorexpr/schedule.h"
#include "torch/csrc/jit/tensorexpr/loopnest.h"
#include "torch/csrc/jit/tensorexpr/tensor.h"
#include <c10/cuda/CUDACachingAllocator.h>

2
test/cpp/tensorexpr/test_expr.cpp
View File

@ -7,7 +7,7 @@
#include "torch/csrc/jit/tensorexpr/function.h"
#include "torch/csrc/jit/tensorexpr/ir.h"
#include "torch/csrc/jit/tensorexpr/ir_printer.h"
#include "torch/csrc/jit/tensorexpr/schedule.h"
#include "torch/csrc/jit/tensorexpr/loopnest.h"
#include "torch/csrc/jit/tensorexpr/tensor.h"
#include <cmath>

2
test/cpp/tensorexpr/test_llvm.cpp
View File

@ -9,7 +9,7 @@
#include "torch/csrc/jit/tensorexpr/ir.h"
#include "torch/csrc/jit/tensorexpr/ir_printer.h"
#include "torch/csrc/jit/tensorexpr/llvm_codegen.h"
#include "torch/csrc/jit/tensorexpr/schedule.h"
#include "torch/csrc/jit/tensorexpr/loopnest.h"
#include "torch/csrc/jit/tensorexpr/tensor.h"
#include <numeric>

2
test/cpp/tensorexpr/test_schedule.cpp → test/cpp/tensorexpr/test_loopnest.cpp
View File

@ -10,7 +10,7 @@
#include "torch/csrc/jit/tensorexpr/function.h"
#include "torch/csrc/jit/tensorexpr/ir.h"
#include "torch/csrc/jit/tensorexpr/ir_printer.h"
#include "torch/csrc/jit/tensorexpr/schedule.h"
#include "torch/csrc/jit/tensorexpr/loopnest.h"
#include "torch/csrc/jit/tensorexpr/tensor.h"
namespace torch {

2
test/cpp/tensorexpr/test_simplify.cpp
View File

@ -4,7 +4,7 @@
#include "torch/csrc/jit/tensorexpr/hash_server.h"
#include "torch/csrc/jit/tensorexpr/ir_simplifier.h"
#include "torch/csrc/jit/tensorexpr/llvm_codegen.h"
#include "torch/csrc/jit/tensorexpr/schedule.h"
#include "torch/csrc/jit/tensorexpr/loopnest.h"
namespace torch {
namespace jit {

2
tools/build_variables.bzl
View File

@ -209,8 +209,8 @@ libtorch_sources = [
"torch/csrc/jit/tensorexpr/kernel.cpp",
"torch/csrc/jit/tensorexpr/llvm_codegen.cpp",
"torch/csrc/jit/tensorexpr/llvm_jit.cpp",
"torch/csrc/jit/tensorexpr/loopnest.cpp",
"torch/csrc/jit/tensorexpr/mem_arena.cpp",
"torch/csrc/jit/tensorexpr/schedule.cpp",
"torch/csrc/jit/tensorexpr/tensor.cpp",
"torch/csrc/jit/tensorexpr/types.cpp",
"torch/csrc/jit/tensorexpr/unique_name_manager.cpp",

2
torch/csrc/jit/tensorexpr/kernel.cpp
View File

@ -4,7 +4,7 @@
#include <torch/csrc/jit/tensorexpr/analysis.h>
#include <torch/csrc/jit/tensorexpr/ir_printer.h>
#include <torch/csrc/jit/tensorexpr/ir_simplifier.h>
#include <torch/csrc/jit/tensorexpr/schedule.h>
#include <torch/csrc/jit/tensorexpr/loopnest.h>
using namespace torch::jit;
using namespace torch::jit::tensorexpr;

3
torch/csrc/jit/tensorexpr/schedule.cpp → torch/csrc/jit/tensorexpr/loopnest.cpp
View File

@ -1,4 +1,4 @@
#include <torch/csrc/jit/tensorexpr/schedule.h>
#include <torch/csrc/jit/tensorexpr/loopnest.h>
#include <queue>
#include <stdexcept>
@ -403,6 +403,7 @@ class FunctionInliner : public IRMutator {
}
}
private:
std::unordered_map<const Var*, const Expr*> inline_mapping_;
std::vector<Function*> funcs_;
std::unordered_set<const Var*> func_var_set_;

0
torch/csrc/jit/tensorexpr/schedule.h → torch/csrc/jit/tensorexpr/loopnest.h
View File