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

irange-ify 11 (#62121)

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

Test Plan: Sandcastle

Reviewed By: ngimel

Differential Revision: D29879701

fbshipit-source-id: 5c51879c88fa6a5790db241c8b33ec0dc4b177ca
This commit is contained in:
Richard Barnes
2021-07-28 13:28:39 -07:00
committed by Facebook GitHub Bot
parent b5867a1b34
commit 9e77113e85
9 changed files with 33 additions and 22 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
torch/csrc/jit/python/init.cpp
View File

@ -98,6 +98,7 @@
#include <torch/csrc/jit/tensorexpr/tensorexpr_init.h>
#include <c10/macros/Export.h>
#include <c10/util/irange.h>
#include <c10/util/signal_handler.h>
#include <caffe2/serialize/inline_container.h>
@ -475,7 +476,7 @@ void initJITBindings(PyObject* module) {
// we want full shape specialization. The alternative would be to
// have a "complete type inference" function in ArguemntSpecCreator.
auto g_inputs = graph->inputs();
for (size_t i = 0; i < inputs.size(); ++i) {
for (const auto i : c10::irange(inputs.size())) {
if (stack[i].isTensor()) {
g_inputs[i]->setType(stack[i].type());
}
@ -491,7 +492,7 @@ void initJITBindings(PyObject* module) {
stack.push_back(toTypeInferredIValue(obj));
}
auto g_inputs = graph->inputs();
for (size_t i = 0; i < inputs.size(); ++i) {
for (const auto i : c10::irange(inputs.size())) {
if (stack[i].isTensor()) {
g_inputs[i]->setType(stack[i].type());
}
@ -1164,7 +1165,7 @@ void initJITBindings(PyObject* module) {
[operations, symbol](py::args args, py::kwargs kwargs) {
std::vector<py::handle> overloaded_args;
size_t total_arg_num = args.size() + kwargs.size();
for (size_t i = 0; i < args.size(); ++i) {
for (const auto i : c10::irange(args.size())) {
is_tensor_and_append_overloaded(
args[i].ptr(), &overloaded_args);
is_tensor_list_and_append_overloaded(
@ -1380,7 +1381,7 @@ void initJITBindings(PyObject* module) {
py::function f = py::cast<py::function>(args[0]);
py::tuple args_tup(args.size() - 1);
for (size_t i = 1; i < args.size(); ++i) {
for (const auto i : c10::irange(1, args.size())) {
args_tup[i - 1] = args[i];
}