Adding all project files
This commit is contained in:
parent
6c9e127bdc
commit
cd4316ad0f
42289 changed files with 8009643 additions and 0 deletions
|
@ -0,0 +1,12 @@
|
|||
#include <c10/macros/Macros.h>
|
||||
#include <c10/core/Device.h>
|
||||
#include <cstdint>
|
||||
|
||||
namespace at::cuda {
|
||||
namespace detail {
|
||||
void init_p2p_access_cache(int64_t num_devices);
|
||||
}
|
||||
|
||||
TORCH_CUDA_CPP_API bool get_p2p_access(c10::DeviceIndex source_dev, c10::DeviceIndex dest_dev);
|
||||
|
||||
} // namespace at::cuda
|
Loading…
Add table
Add a link
Reference in a new issue