Skip to content

Commit a38df36

Browse files
jeffbolznvmglambda
authored andcommitted
vulkan: print shared memory size (ggml-org#11719)
1 parent b5953fa commit a38df36

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

ggml/src/ggml-vulkan/ggml-vulkan.cpp

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -2780,8 +2780,9 @@ static void ggml_vk_print_gpu_info(size_t idx) {
27802780
std::string matrix_cores = coopmat2_support ? "NV_coopmat2" : coopmat_support ? "KHR_coopmat" : "none";
27812781

27822782
std::string device_name = props2.properties.deviceName.data();
2783-
GGML_LOG_DEBUG("ggml_vulkan: %zu = %s (%s) | uma: %d | fp16: %d | warp size: %zu | matrix cores: %s\n",
2784-
idx, device_name.c_str(), driver_props.driverName.data(), uma, fp16, subgroup_size, matrix_cores.c_str());
2783+
GGML_LOG_DEBUG("ggml_vulkan: %zu = %s (%s) | uma: %d | fp16: %d | warp size: %zu | shared memory: %d | matrix cores: %s\n",
2784+
idx, device_name.c_str(), driver_props.driverName.data(), uma, fp16, subgroup_size,
2785+
props2.properties.limits.maxComputeSharedMemorySize, matrix_cores.c_str());
27852786

27862787
if (props2.properties.deviceType == vk::PhysicalDeviceType::eCpu) {
27872788
GGML_LOG_DEBUG("ggml_vulkan: Warning: Device type is CPU. This is probably not the device you want.\n");

0 commit comments

Comments
 (0)