|
4 | 4 | from .constants import (API_BLAS, API_C10, API_CAFFE2, API_DRIVER, API_FFT,
|
5 | 5 | API_PYTORCH, API_RAND, API_ROCTX, API_RTC, API_RUNTIME,
|
6 | 6 | API_SPECIAL, API_ROCMSMI, CONV_CACHE, CONV_CONTEXT, CONV_D3D9,
|
7 |
| - CONV_D3D10, CONV_D3D11, CONV_DEF, CONV_DEVICE, |
| 7 | + API_PYT_EXT, CONV_D3D10, CONV_D3D11, CONV_DEF, CONV_DEVICE, |
8 | 8 | CONV_DEVICE_FUNC, CONV_EGL, CONV_ERROR, CONV_EVENT,
|
9 | 9 | CONV_EXEC, CONV_GL, CONV_GRAPHICS, CONV_INCLUDE,
|
10 | 10 | CONV_INCLUDE_CUDA_MAIN_H, CONV_INIT, CONV_JIT,
|
|
631 | 631 | ("nvtx3/nvtx3.hpp", ("roctracer/roctx.h", CONV_INCLUDE, API_ROCTX)),
|
632 | 632 | ("nvToolsExt.h", ("roctracer/roctx.h", CONV_INCLUDE, API_ROCTX)),
|
633 | 633 | ("nvml.h", ("rocm_smi/rocm_smi.h", CONV_INCLUDE, API_ROCMSMI)),
|
| 634 | + ("tensorpipe/tensorpipe_cuda.h", ("tensorpipe/tensorpipe_hip.h", CONV_INCLUDE, API_PYT_EXT)), |
634 | 635 | ]
|
635 | 636 | )
|
636 | 637 |
|
|
0 commit comments