mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-20 21:14:14 +08:00
d568c7d966
Summary: Add an explanation string to malformed_input exceptions thrown inside jit/tensorexpr to aid in debugging issues. Pull Request resolved: https://github.com/pytorch/pytorch/pull/35891 Differential Revision: D20822306 Pulled By: nickgg fbshipit-source-id: ce153a05218f2a4da5ecf5f1a5dc439070c96e55
72 lines
2.3 KiB
C++
72 lines
2.3 KiB
C++
#pragma once
|
|
|
|
#include <torch/csrc/WindowsTorchApiMacro.h>
|
|
|
|
#include <sstream>
|
|
#include <stdexcept>
|
|
|
|
// Forward declarations of types
|
|
namespace torch {
|
|
namespace jit {
|
|
namespace tensorexpr {
|
|
class Expr;
|
|
class Stmt;
|
|
} // namespace tensorexpr
|
|
} // namespace jit
|
|
} // namespace torch
|
|
|
|
// Forward declarations of functions
|
|
namespace std {
|
|
TORCH_API std::string to_string(const torch::jit::tensorexpr::Expr*);
|
|
TORCH_API std::string to_string(const torch::jit::tensorexpr::Stmt*);
|
|
} // namespace std
|
|
|
|
namespace torch {
|
|
namespace jit {
|
|
namespace tensorexpr {
|
|
|
|
class unsupported_dtype : public std::runtime_error {
|
|
public:
|
|
explicit unsupported_dtype() : std::runtime_error("UNSUPPORTED DTYPE") {}
|
|
explicit unsupported_dtype(const std::string& err)
|
|
: std::runtime_error("UNSUPPORTED DTYPE: " + err) {}
|
|
};
|
|
|
|
class out_of_range_index : public std::runtime_error {
|
|
public:
|
|
explicit out_of_range_index() : std::runtime_error("OUT OF RANGE INDEX") {}
|
|
explicit out_of_range_index(const std::string& err)
|
|
: std::runtime_error("OUT OF RANGE INDEX: " + err) {}
|
|
};
|
|
|
|
class unimplemented_lowering : public std::runtime_error {
|
|
public:
|
|
explicit unimplemented_lowering()
|
|
: std::runtime_error("UNIMPLEMENTED LOWERING") {}
|
|
explicit unimplemented_lowering(const Expr* expr)
|
|
: std::runtime_error("UNIMPLEMENTED LOWERING: " + std::to_string(expr)) {}
|
|
explicit unimplemented_lowering(const Stmt* stmt)
|
|
: std::runtime_error("UNIMPLEMENTED LOWERING: " + std::to_string(stmt)) {}
|
|
};
|
|
|
|
class malformed_input : public std::runtime_error {
|
|
public:
|
|
explicit malformed_input() : std::runtime_error("MALFORMED INPUT") {}
|
|
explicit malformed_input(const std::string& err)
|
|
: std::runtime_error("MALFORMED INPUT: " + err) {}
|
|
explicit malformed_input(const Expr* expr)
|
|
: std::runtime_error("MALFORMED INPUT: " + std::to_string(expr)) {}
|
|
explicit malformed_input(const std::string& err, const Expr* expr)
|
|
: std::runtime_error(
|
|
"MALFORMED INPUT: " + err + " - " + std::to_string(expr)) {}
|
|
explicit malformed_input(const Stmt* stmt)
|
|
: std::runtime_error("MALFORMED INPUT: " + std::to_string(stmt)) {}
|
|
explicit malformed_input(const std::string& err, const Stmt* stmt)
|
|
: std::runtime_error(
|
|
"MALFORMED INPUT: " + err + " - " + std::to_string(stmt)) {}
|
|
};
|
|
|
|
} // namespace tensorexpr
|
|
} // namespace jit
|
|
} // namespace torch
|