Skip to content

Commit ac7f0ab

Browse files
authored
[ORO-0] Get global memory size for CUDA (#44)
1 parent 6ba85ab commit ac7f0ab

1 file changed

Lines changed: 2 additions & 1 deletion

File tree

Orochi/Orochi.cpp

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -296,11 +296,12 @@ oroError OROAPI oroGetDeviceProperties(oroDeviceProp* props, oroDevice dev)
296296
props->memPitch = p.memPitch;
297297
props->clockRate = p.clockRate;
298298

299+
cu2oro( cuDeviceTotalMem( &props->totalGlobalMem, deviceId ) );
300+
299301
e = cuDeviceGetAttribute( &props->pciDomainID, CU_DEVICE_ATTRIBUTE_PCI_DOMAIN_ID, deviceId );
300302
e = cuDeviceGetAttribute(&props->pciBusID, CU_DEVICE_ATTRIBUTE_PCI_BUS_ID, deviceId);
301303
e = cuDeviceGetAttribute(&props->pciDeviceID, CU_DEVICE_ATTRIBUTE_PCI_DEVICE_ID, deviceId);
302304
e = cuDeviceGetAttribute(&props->multiProcessorCount, CU_DEVICE_ATTRIBUTE_MULTIPROCESSOR_COUNT, deviceId);
303-
// props->totalGlobalMem = p.totalGlobalMem;? todo. DeviceTotalMem instead?
304305
e = cuDeviceGetAttribute( &props->warpSize, CU_DEVICE_ATTRIBUTE_WARP_SIZE, deviceId );
305306
e = cuDeviceGetAttribute( (int*)&props->textureAlignment, CU_DEVICE_ATTRIBUTE_TEXTURE_ALIGNMENT, deviceId );
306307
e = cuDeviceGetAttribute( &props->kernelExecTimeoutEnabled, CU_DEVICE_ATTRIBUTE_KERNEL_EXEC_TIMEOUT, deviceId );

0 commit comments

Comments
 (0)