/usr/local/lib64/python3.6/site-packages/torch/include/torch/csrc/jit/tensorexpr
Edit: /usr/local/lib64/python3.6/site-packages/torch/include/torch/csrc/jit/tensorexpr/llvm_jit.h (1965B)
#pragma once
#ifdef TORCH_ENABLE_LLVM
#include
#include
#include
#include
#include
#include
#include
#include
#include
namespace torch {
namespace jit {
namespace tensorexpr {
extern "C" {
void DispatchParallel(
int8_t* func,
int64_t start,
int64_t stop,
int8_t* packed_data) noexcept;
}
inline std::string formatError(llvm::Error&& err, const char* msg) {
static constexpr char* defaultErrorMsg = "Unexpected failure in LLVM JIT";
std::string errorMsg(msg ? msg : defaultErrorMsg);
llvm::raw_string_ostream ss(errorMsg);
ss << ": " << err;
return ss.str();
}
template
T assertSuccess(llvm::Expected valOrErr, const char* msg = nullptr) {
TORCH_INTERNAL_ASSERT(valOrErr, formatError(valOrErr.takeError(), msg));
return std::move(*valOrErr);
}
inline void assertSuccess(llvm::Error err, const char* msg = nullptr) {
TORCH_INTERNAL_ASSERT(!err, formatError(std::move(err), msg));
}
} // namespace tensorexpr
} // namespace jit
} // namespace torch
namespace llvm {
namespace orc {
class PytorchLLVMJITImpl;
class TORCH_API PytorchLLVMJIT {
public:
PytorchLLVMJIT(
c10::optional triple,
c10::optional cpu,
c10::optional attrs);
~PytorchLLVMJIT();
void addModule(std::unique_ptr M, std::unique_ptr C);
JITSymbol findSymbol(const std::string Name);
bool hasSymbol(const std::string& Name);
TargetMachine& getTargetMachine();
const DataLayout& getDataLayout();
private:
// Use the PImpl idiom here to hide the no-rtti parts of the JIT structure.
std::unique_ptr impl_;
};
} // end namespace orc
} // end namespace llvm
#endif // ENABLE LLVM