devtools/eigen
1
0
Fork 0
mirror of https://gitlab.com/libeigen/eigen.git synced 2026-04-10 11:34:33 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fix bad static access for TensorDeviceGpu.

Browse Source 
libeigen/eigen!2111
This commit is contained in:
Antonio Sánchez
2026-01-20 19:12:53 +00:00
parent c09151a2c2
commit f46a2c561e
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
unsupported/Eigen/CXX11/src/Tensor/TensorDeviceGpu.h
View File

@@ -45,9 +45,9 @@ class StreamInterface {
class GpuDeviceProperties { class GpuDeviceProperties {
public: public:
static const GpuDeviceProperties& instance() { static const GpuDeviceProperties& instance() {
static const GpuDeviceProperties& kInstance = *new GpuDeviceProperties(); static const GpuDeviceProperties* kInstance = new GpuDeviceProperties();
return kInstance; return *kInstance;
} }
EIGEN_STRONG_INLINE const gpuDeviceProp_t& get(int device) const { return device_properties_[device]; } EIGEN_STRONG_INLINE const gpuDeviceProp_t& get(int device) const { return device_properties_[device]; }
@@ -64,7 +64,7 @@ class GpuDeviceProperties {
} }
std::vector<gpuDeviceProp_t> device_properties(num_devices); std::vector<gpuDeviceProp_t> device_properties(num_devices);
for (int i = 0; i < num_devices; ++i) { for (int i = 0; i < num_devices; ++i) {
status = gpuGetDeviceProperties(&device_properties_[i], i); status = gpuGetDeviceProperties(&device_properties[i], i);
if (status != gpuSuccess) { if (status != gpuSuccess) {
std::cerr << "Failed to initialize GPU device #" << i << ": " << gpuGetErrorString(status) << std::endl; std::cerr << "Failed to initialize GPU device #" << i << ": " << gpuGetErrorString(status) << std::endl;
gpu_assert(status == gpuSuccess); gpu_assert(status == gpuSuccess);