1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
|
#pragma once
#include <Python.h>
#include <torch/csrc/profiler/collection.h>
#include <torch/csrc/profiler/python/pybind.h>
namespace pybind11 {
namespace detail {
using torch::profiler::impl::StorageImplData;
using torch::profiler::impl::TensorID;
using torch::profiler::impl::TensorImplAddress;
template <>
struct type_caster<StorageImplData>
: public strong_pointer_type_caster<StorageImplData> {};
template <>
struct type_caster<TensorImplAddress>
: public strong_pointer_type_caster<TensorImplAddress> {};
template <>
struct type_caster<TensorID> : public strong_uint_type_caster<TensorID> {};
} // namespace detail
} // namespace pybind11
namespace torch {
namespace profiler {
void initPythonBindings(PyObject* module);
} // namespace profiler
} // namespace torch
|
