Why Gemfury?
Push, build, and install
RubyGems
npm packages
Python packages
Maven artifacts
PHP packages
Go Modules
Debian packages
RPM packages
NuGet packages
duality-group / torch python
Repository URL to install this package:
|
Version:
1.12.1+cpu ▾
|
#pragma once
#include <ATen/cuda/Exceptions.h>
#include <cuda.h>
#include <cuda_runtime.h>
namespace at {
namespace cuda {
inline Device getDeviceFromPtr(void* ptr) {
cudaPointerAttributes attr;
AT_CUDA_CHECK(cudaPointerGetAttributes(&attr, ptr));
return {DeviceType::CUDA, static_cast<DeviceIndex>(attr.device)};
}
}} // namespace at::cuda