mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-21 13:44:15 +08:00
40fb738197
Fixes #ISSUE_NUMBER Pull Request resolved: https://github.com/pytorch/pytorch/pull/140236 Approved by: https://github.com/ezyang
32 lines
998 B
C++
32 lines
998 B
C++
#pragma once
|
|
|
|
#include <Python.h>
|
|
|
|
#include <torch/csrc/profiler/collection.h>
|
|
#include <torch/csrc/profiler/python/pybind.h>
|
|
|
|
namespace pybind11::detail {
|
|
using torch::profiler::impl::TensorID;
|
|
|
|
#define STRONG_POINTER_TYPE_CASTER(T) \
|
|
template <> \
|
|
struct type_caster<T> : public strong_pointer_type_caster<T> {};
|
|
|
|
STRONG_POINTER_TYPE_CASTER(torch::profiler::impl::StorageImplData)
|
|
STRONG_POINTER_TYPE_CASTER(torch::profiler::impl::AllocationID)
|
|
STRONG_POINTER_TYPE_CASTER(torch::profiler::impl::TensorImplAddress)
|
|
STRONG_POINTER_TYPE_CASTER(torch::profiler::impl::PyModuleSelf)
|
|
STRONG_POINTER_TYPE_CASTER(torch::profiler::impl::PyModuleCls)
|
|
STRONG_POINTER_TYPE_CASTER(torch::profiler::impl::PyOptimizerSelf)
|
|
#undef STRONG_POINTER_TYPE_CASTER
|
|
|
|
template <>
|
|
struct type_caster<TensorID> : public strong_uint_type_caster<TensorID> {};
|
|
} // namespace pybind11::detail
|
|
|
|
namespace torch::profiler {
|
|
|
|
void initPythonBindings(PyObject* module);
|
|
|
|
} // namespace torch::profiler
|