mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-07 12:21:27 +01:00
8f0603b128
Summary: * adds TORCH_API and AT_CUDA_API in places * refactor code generation Python logic to separate caffe2/torch outputs * fix hip and asan * remove profiler_cuda from hip * fix gcc warnings for enums * Fix PythonOp::Kind Pull Request resolved: https://github.com/pytorch/pytorch/pull/19554 Differential Revision: D15082727 Pulled By: kostmo fbshipit-source-id: 83a8a99717f025ab44b29608848928d76b3147a4
31 lines
795 B
C++
31 lines
795 B
C++
#include <torch/csrc/WindowsTorchApiMacro.h>
|
|
#include <string>
|
|
#include <unordered_map>
|
|
|
|
#include <torch/csrc/WindowsTorchApiMacro.h>
|
|
|
|
namespace torch {
|
|
namespace jit {
|
|
|
|
struct Graph;
|
|
struct Value;
|
|
|
|
namespace script {
|
|
|
|
// \brief Parse IR from \p STR constructing the corresponding IR in\ GRAPH.
|
|
TORCH_API void parseIR(const std::string& str, torch::jit::Graph* graph);
|
|
|
|
/** \brief Parse IR from \p STR constructing the corresponding IR in\ GRAPH.
|
|
*
|
|
* \p VMAP is filled with String to Value pairs allowing to index Values in the
|
|
* newly created graph by their name in the original IR string.
|
|
*/
|
|
TORCH_API void parseIR(
|
|
const std::string& str,
|
|
torch::jit::Graph* graph,
|
|
std::unordered_map<std::string, Value*>& vmap);
|
|
|
|
} // namespace script
|
|
} // namespace jit
|
|
} // namespace torch
|