mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-06 12:20:52 +01:00
713e706618
Summary: There are still a few work to be done: - Move logging and unify AT_WARN with LOG(ERROR). - A few header files are still being plumbed through, need cleaning. - caffe2::EnforceNotMet aliasing is not done yet. - need to unify the macros. See c10/util/Exception.h This is mainly a codemod and not causing functional changes. If you find your job failing and trace back to this diff, usually it can be fixed by the following approaches: (1) add //caffe2/c10:c10 to your dependency (or transitive dependency). (2) change objects such as at::Error, at::Optional to the c10 namespace. (3) change functions to the c10 namespace. Especially, caffe2::MakeString is not overridden by the unified c10::str function. Nothing else changes. Please kindly consider not reverting this diff - it involves multiple rounds of rebasing and the fix is usually simple. Contact jiayq@ or AI Platform Dev for details. Pull Request resolved: https://github.com/pytorch/pytorch/pull/12354 Reviewed By: orionr Differential Revision: D10238910 Pulled By: Yangqing fbshipit-source-id: 7794d5bf2797ab0ca6ebaccaa2f7ebbd50ff8f32
44 lines
1.0 KiB
C++
44 lines
1.0 KiB
C++
#include "c10/util/StringUtil.h"
|
|
#include "c10/util/Exception.h"
|
|
|
|
#include <cstring>
|
|
#include <string>
|
|
|
|
namespace c10 {
|
|
|
|
namespace detail {
|
|
|
|
std::string StripBasename(const std::string& full_path) {
|
|
const char kSeparator = '/';
|
|
size_t pos = full_path.rfind(kSeparator);
|
|
if (pos != std::string::npos) {
|
|
return full_path.substr(pos + 1, std::string::npos);
|
|
} else {
|
|
return full_path;
|
|
}
|
|
}
|
|
|
|
} // namespace detail
|
|
|
|
std::ostream& operator<<(std::ostream& out, const SourceLocation& loc) {
|
|
out << loc.function << " at " << loc.file << ":" << loc.line;
|
|
return out;
|
|
}
|
|
|
|
size_t ReplaceAll(std::string& s, const char* from, const char* to) {
|
|
AT_CHECK(from && *from, "");
|
|
AT_CHECK(to, "");
|
|
|
|
size_t numReplaced = 0;
|
|
std::string::size_type lenFrom = std::strlen(from);
|
|
std::string::size_type lenTo = std::strlen(to);
|
|
for (auto pos = s.find(from); pos != std::string::npos;
|
|
pos = s.find(from, pos + lenTo)) {
|
|
s.replace(pos, lenFrom, to);
|
|
numReplaced++;
|
|
}
|
|
return numReplaced;
|
|
}
|
|
|
|
} // namespace c10
|