review.tizen.org Git - platform/upstream/pytorch.git/commit

projects / platform / upstream / pytorch.git / commit

author	Bert Maher <bertrand@fb.com>
	Sun, 29 Aug 2021 02:57:10 +0000 (19:57 -0700)
committer	Facebook GitHub Bot <facebook-github-bot@users.noreply.github.com>
	Sun, 29 Aug 2021 02:59:47 +0000 (19:59 -0700)
commit	2e6221a232d39917e2736b248c53fa85dfb8986e
tree	7040133c9d3f47ba0ded53c40891542a9d4be4d5	tree \| snapshot
parent	405c15516c3a052fbad33fa8fcb6cde84ebc97d8	commit \| diff

[nnc] Make 64-bit dimensions work (#64077)

Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/64077

We were assuming kernel dimensions fit in 32 bits (the old fuser made
this assumption too), but we should be able to support 64.
ghstack-source-id: 136933272

Test Plan: unit tests; new IR level test with huge sizes

Reviewed By: ZolotukhinM

Differential Revision: D30596689

fbshipit-source-id: 23b7e393a2ebaecb0c391a6b1f0c4b05a98bcc94

23 files changed:

test/cpp/tensorexpr/test_kernel.cpp		diff \| blob \| history
test/cpp/tensorexpr/test_llvm.cpp		diff \| blob \| history
test/cpp/tensorexpr/test_loopnest.cpp		diff \| blob \| history
test/cpp/tensorexpr/test_reductions.cpp		diff \| blob \| history
torch/csrc/jit/tensorexpr/block_codegen.cpp		diff \| blob \| history
torch/csrc/jit/tensorexpr/bounds_inference.cpp		diff \| blob \| history
torch/csrc/jit/tensorexpr/bounds_overlap.cpp		diff \| blob \| history
torch/csrc/jit/tensorexpr/cuda_codegen.cpp		diff \| blob \| history
torch/csrc/jit/tensorexpr/eval.cpp		diff \| blob \| history
torch/csrc/jit/tensorexpr/eval.h		diff \| blob \| history
torch/csrc/jit/tensorexpr/expr.h		diff \| blob \| history
torch/csrc/jit/tensorexpr/ir.cpp		diff \| blob \| history
torch/csrc/jit/tensorexpr/ir.h		diff \| blob \| history
torch/csrc/jit/tensorexpr/ir_printer.cpp		diff \| blob \| history
torch/csrc/jit/tensorexpr/ir_simplifier.cpp		diff \| blob \| history
torch/csrc/jit/tensorexpr/kernel.cpp		diff \| blob \| history
torch/csrc/jit/tensorexpr/kernel.h		diff \| blob \| history
torch/csrc/jit/tensorexpr/llvm_codegen.cpp		diff \| blob \| history
torch/csrc/jit/tensorexpr/llvm_jit.h		diff \| blob \| history
torch/csrc/jit/tensorexpr/loopnest.cpp		diff \| blob \| history
torch/csrc/jit/tensorexpr/mem_dependency_checker.cpp		diff \| blob \| history
torch/csrc/jit/tensorexpr/registerizer.cpp		diff \| blob \| history
torch/csrc/jit/tensorexpr/tensor.cpp		diff \| blob \| history

Domain: Machine Learning / ML Framework;

RSS Atom