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

[Reland2] fix missing-prototypes warnings in torch_cpu (Part 4) (#102228)

Browse Source 
This PR relands the changes introduced in PR https://github.com/pytorch/pytorch/pull/100849. The old PR turnd nnc_* functions into  static. We now add declarations for them and hope that inter builds will pass.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/102228
Approved by: https://github.com/albanD
This commit is contained in:
cyy
2023-06-02 22:04:39 +00:00
committed by PyTorch MergeBot
parent 86c7652503
commit 77f2883c41
138 changed files with 577 additions and 773 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
torch/csrc/jit/runtime/static/impl.cpp
View File

@ -76,7 +76,7 @@ bool allArgsAreTensors(const Node* node) {
// These are rarely-used ops. Disallowing them typically eliminates
// corner cases in graph optimizations, allowing for more aggressive
// optimizations and better performance.
bool isUnsupportedOp(const Node* node) {
static bool isUnsupportedOp(const Node* node) {
auto kind = node->kind();
if (kind != aten::__is__ && kind != aten::__isnot__) {
return false;
@ -1587,7 +1587,7 @@ float BlockRunner::benchmark_model(
(static_cast<float>(main_runs) * static_cast<float>(args_list.size()));
}
bool display_ivalue(const IValue& iv) {
static bool display_ivalue(const IValue& iv) {
if (iv.isTensor()) {
std::cout << "Tensor " << iv.toTensor().toString() << " {";
const auto dims = iv.toTensor().sizes();
@ -1622,7 +1622,7 @@ bool display_ivalue(const IValue& iv) {
return false;
}
void display_pnode_info(const ProcessedNode& pnode) {
static void display_pnode_info(const ProcessedNode& pnode) {
pnode.node()->print(std::cout, 0, nullptr, false);
const auto num_inputs = static_cast<uint32_t>(pnode.num_inputs());
for (const auto i : c10::irange(num_inputs)) {