// #include <THC/THC.h> //"torch"库从1.11.0版本开始，就不再支持使"THC/THC.h"头文件了
// https://www.coder.work/article/7897377
// 替换为下面两个.h文件。同时，注释掉extern THCState *state; 就可以成功编译
#include <ATen/cuda/CUDAContext.h>
#include <ATen/cuda/CUDAEvent.h>

// extern THCState *state;