torch/csrc/dynamo/cpp_shim.h

#pragma once

#ifdef __cplusplus
extern "C" {
#endif

struct _PytorchRecordFunctionState;
typedef struct _PytorchRecordFunctionState _PytorchRecordFunctionState;

_PytorchRecordFunctionState* _pytorch_record_function_enter(const char* name);
void _pytorch_record_function_exit(_PytorchRecordFunctionState* state);

#ifdef __cplusplus
} // extern "C"
#endif