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

Remaining replacement of c10::stoi with std::stoi (#109482)

Browse Source 
PR #109179 replaced c10::stoi with std::stoi. However, there were some files forgotten to change. This patch fixes them.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/109482
Approved by: https://github.com/vfdev-5, https://github.com/Skylion007
This commit is contained in:
cyy
2023-09-18 16:05:05 +00:00
committed by PyTorch MergeBot
parent 6ffa59031a
commit 5d5990fc49
5 changed files with 5 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
binaries/benchmark_helper.cc
View File

@ -163,7 +163,7 @@ int loadInput(
vector<string> input_dims_str = caffe2::split(',', input_dims_list[i]); vector<string> input_dims_str = caffe2::split(',', input_dims_list[i]);
vector<int> input_dims; vector<int> input_dims;
for (const string& s : input_dims_str) { for (const string& s : input_dims_str) {
input_dims.push_back(c10::stoi(s)); input_dims.push_back(std::stoi(s));
} }
caffe2::Blob* blob = workspace->GetBlob(input_names[i]); caffe2::Blob* blob = workspace->GetBlob(input_names[i]);
if (blob == nullptr) { if (blob == nullptr) {

2
binaries/compare_models_torch.cc
View File

@ -170,7 +170,7 @@ std::vector<c10::IValue> create_inputs(
std::vector<int64_t> input_dims; std::vector<int64_t> input_dims;
input_dims.reserve(input_dims_str.size()); input_dims.reserve(input_dims_str.size());
for (const auto& s : input_dims_str) { for (const auto& s : input_dims_str) {
input_dims.push_back(c10::stoi(s)); input_dims.push_back(std::stoi(s));
} }
at::ScalarType input_type; at::ScalarType input_type;

2
binaries/speed_benchmark.cc
View File

@ -127,7 +127,7 @@ int main(int argc, char** argv) {
vector<string> input_dims_str = caffe2::split(',', input_dims_list[i]); vector<string> input_dims_str = caffe2::split(',', input_dims_list[i]);
vector<int> input_dims; vector<int> input_dims;
for (const string& s : input_dims_str) { for (const string& s : input_dims_str) {
input_dims.push_back(c10::stoi(s)); input_dims.push_back(std::stoi(s));
} }
caffe2::Blob* blob = workspace->GetBlob(input_names[i]); caffe2::Blob* blob = workspace->GetBlob(input_names[i]);
if (blob == nullptr) { if (blob == nullptr) {

2
binaries/speed_benchmark_torch.cc
View File

@ -119,7 +119,7 @@ std::vector<c10::IValue> create_inputs() {
auto input_dims_str = split(',', input_dims_list[i]); auto input_dims_str = split(',', input_dims_list[i]);
std::vector<int64_t> input_dims; std::vector<int64_t> input_dims;
for (const auto& s : input_dims_str) { for (const auto& s : input_dims_str) {
input_dims.push_back(c10::stoi(s)); input_dims.push_back(std::stoi(s));
} }
at::ScalarType input_type; at::ScalarType input_type;

2
ios/TestApp/TestApp/Benchmark.mm
View File

@ -60,7 +60,7 @@ static int iter = 10;
auto input_dims_str = caffe2::split(',', input_dims_list[i]); auto input_dims_str = caffe2::split(',', input_dims_list[i]);
std::vector<int64_t> input_dims; std::vector<int64_t> input_dims;
for (const auto& s : input_dims_str) { for (const auto& s : input_dims_str) {
input_dims.push_back(c10::stoi(s)); input_dims.push_back(std::stoi(s));
} }
if (input_type_list[i] == "float") { if (input_type_list[i] == "float") {
inputs.push_back(torch::ones(input_dims, at::ScalarType::Float)); inputs.push_back(torch::ones(input_dims, at::ScalarType::Float));