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

Revert D32175958: Adding Custom Rules to Device Propagation

Browse Source 
Test Plan: revert-hammer

Differential Revision:
D32175958 (853298481b)

Original commit changeset: 26a9ef41e10a

fbshipit-source-id: adcc70687b5b454f358b5446bed2c06d04e61435
This commit is contained in:
Alban Desmaison
2021-11-05 07:01:30 -07:00
committed by Facebook GitHub Bot
parent b1ac7f51a1
commit 38af37f409
3 changed files with 2 additions and 85 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
torch/csrc/jit/ir/ir.cpp
View File

@ -14,7 +14,6 @@
#include <algorithm>
#include <iostream>
#include <memory>
#include <set>
#include <sstream>
#include <string>
@ -2304,15 +2303,6 @@ OperatorSet::OperatorSet(std::initializer_list<const char*> sig_literals) {
}
}
std::vector<std::shared_ptr<Operator>> OperatorSet::getOps() const {
std::vector<std::shared_ptr<Operator>> result;
for (const auto& kv : ops) {
auto ops_for_symbol = kv.second;
result.insert(result.end(), ops_for_symbol.begin(), ops_for_symbol.end());
}
return result;
}
bool Node::isMemberOf(const OperatorSet& os) const {
auto it = os.ops.find(kind());
if (it == os.ops.end()) {