mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-07 12:21:27 +01:00
ad8aef0f98
Mostly in torch/csrc/jit/runtime and in `ATen/cuda/` Pull Request resolved: https://github.com/pytorch/pytorch/pull/110314 Approved by: https://github.com/seemethere
31 lines
917 B
C++
31 lines
917 B
C++
#include <torch/csrc/jit/runtime/jit_exception.h>
|
|
|
|
namespace torch::jit {
|
|
|
|
static thread_local std::string caughtOriginalMsg = "";
|
|
static thread_local std::string caughtPythonClassName = "";
|
|
|
|
JITException::JITException(
|
|
const std::string& msg,
|
|
c10::optional<std::string> python_class_name,
|
|
c10::optional<std::string> original_msg)
|
|
: std::runtime_error(msg),
|
|
python_class_name_(std::move(python_class_name)),
|
|
original_msg_(std::move(original_msg)) {}
|
|
|
|
const std::string& JITException::getCaughtOriginalMsg() {
|
|
return caughtOriginalMsg;
|
|
}
|
|
const std::string& JITException::getCaughtPythonClassName() {
|
|
return caughtPythonClassName;
|
|
}
|
|
void JITException::setCaughtOriginalMsg(const std::string& msg) {
|
|
caughtOriginalMsg = msg;
|
|
}
|
|
void JITException::setCaughtPythonClassName(
|
|
const std::string& pythonClassName) {
|
|
caughtPythonClassName = pythonClassName;
|
|
}
|
|
|
|
} // namespace torch::jit
|