mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-07 12:21:27 +01:00
4cb534f92e
Summary:
This is an automatic change generated by the following script:
```
#!/usr/bin/env python3
from subprocess import check_output, check_call
import os
def get_compiled_files_list():
import json
with open("build/compile_commands.json") as f:
data = json.load(f)
files = [os.path.relpath(node['file']) for node in data]
for idx, fname in enumerate(files):
if fname.startswith('build/') and fname.endswith('.DEFAULT.cpp'):
files[idx] = fname[len('build/'):-len('.DEFAULT.cpp')]
return files
def run_clang_tidy(fname):
check_call(["python3", "tools/clang_tidy.py", "-c", "build", "-x", fname,"-s"])
changes = check_output(["git", "ls-files", "-m"])
if len(changes) == 0:
return
check_call(["git", "commit","--all", "-m", f"NOLINT stubs for {fname}"])
def main():
git_files = check_output(["git", "ls-files"]).decode("ascii").split("\n")
compiled_files = get_compiled_files_list()
for idx, fname in enumerate(git_files):
if fname not in compiled_files:
continue
if fname.startswith("caffe2/contrib/aten/"):
continue
print(f"[{idx}/{len(git_files)}] Processing {fname}")
run_clang_tidy(fname)
if __name__ == "__main__":
main()
```
Pull Request resolved: https://github.com/pytorch/pytorch/pull/56892
Reviewed By: H-Huang
Differential Revision: D27991944
Pulled By: malfet
fbshipit-source-id: 5415e1eb2c1b34319a4f03024bfaa087007d7179
79 lines
1.9 KiB
C++
79 lines
1.9 KiB
C++
#include "smart_tensor_printer.h"
|
|
|
|
#include "caffe2/core/operator.h"
|
|
|
|
namespace caffe2 {
|
|
|
|
namespace {
|
|
|
|
// Since DispatchHelper doesn't support passing arguments through the call()
|
|
// method to DoRunWithType we have to create an object that will hold these
|
|
// arguments explicitly.
|
|
struct ProxyPrinter {
|
|
template <typename T>
|
|
bool DoRunWithType() {
|
|
tensorPrinter->Print<T>(*tensor);
|
|
return true;
|
|
}
|
|
|
|
void Print() {
|
|
// Pulled in printable types from caffe2/core/types.cc
|
|
// Unfortunately right now one has to add them by hand
|
|
DispatchHelper<TensorTypes<
|
|
float,
|
|
int,
|
|
std::string,
|
|
bool,
|
|
uint8_t,
|
|
int8_t,
|
|
uint16_t,
|
|
int16_t,
|
|
int64_t,
|
|
double,
|
|
char>>::call(this, tensor->dtype());
|
|
}
|
|
|
|
const Tensor* tensor;
|
|
TensorPrinter* tensorPrinter;
|
|
};
|
|
} // namespace
|
|
|
|
SmartTensorPrinter::SmartTensorPrinter(const std::string& tensor_name)
|
|
: tensorPrinter_(tensor_name) {}
|
|
|
|
SmartTensorPrinter::SmartTensorPrinter(
|
|
const std::string& tensor_name,
|
|
const std::string& file_name)
|
|
: tensorPrinter_(tensor_name, file_name) {}
|
|
|
|
SmartTensorPrinter::SmartTensorPrinter(
|
|
const std::string& tensor_name,
|
|
const std::string& file_name,
|
|
int limit)
|
|
: tensorPrinter_(tensor_name, file_name, limit) {}
|
|
|
|
void SmartTensorPrinter::Print(const Tensor& tensor) {
|
|
// NOLINTNEXTLINE(cppcoreguidelines-pro-type-member-init)
|
|
ProxyPrinter printer;
|
|
|
|
printer.tensor = &tensor;
|
|
printer.tensorPrinter = &tensorPrinter_;
|
|
printer.Print();
|
|
}
|
|
|
|
SmartTensorPrinter& SmartTensorPrinter::DefaultTensorPrinter() {
|
|
// TODO(janusz): thread_local does not work under mac.
|
|
#if defined(__APPLE__)
|
|
CAFFE_THROW(
|
|
"SmartTensorPrinter does not work on mac yet due to thread_local.");
|
|
#else
|
|
static thread_local SmartTensorPrinter printer;
|
|
return printer;
|
|
#endif
|
|
}
|
|
|
|
void SmartTensorPrinter::PrintTensor(const Tensor& tensor) {
|
|
DefaultTensorPrinter().Print(tensor);
|
|
}
|
|
} // namespace caffe2
|