gaspersic/triton
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

cu_device::max_shared_memory() now returns max dynamic shared memory size (#127)

Browse Source
This commit is contained in:
daadaada
2021-06-22 23:08:48 +08:00
committed by Philippe Tillet
parent d8d6b715c8
commit 0b05e06c0d
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
lib/driver/device.cc
View File

@@ -108,7 +108,7 @@ size_t cu_device::max_threads_per_block() const {
// maximum amount of shared memory per block
size_t cu_device::max_shared_memory() const {
return cuGetInfo<CU_DEVICE_ATTRIBUTE_MAX_SHARED_MEMORY_PER_BLOCK>();
return cuGetInfo<CU_DEVICE_ATTRIBUTE_MAX_SHARED_MEMORY_PER_BLOCK_OPTIN>();
}
// warp size