pytorch/torch/csrc/dynamo/cpp_shim.h
2024-09-20 17:10:27 +00:00

16 lines
357 B
C

#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