Skip to content

Commit 9036e7a

Browse files
committed
Apply map to all NAVI1* gpus
1 parent 04100e8 commit 9036e7a

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

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

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1424,10 +1424,10 @@ static bool ggml_vk_matmul_shmem_support(const vk_device& device, const std::vec
14241424
}
14251425

14261426
// Define a configuration map per GPU.
1427-
// Outer key: GPU identifier (e.g. "RX 5700").
1427+
// Outer key: GPU identifier (e.g. "NAVI1").
14281428
// Inner map: key is pipeline name; value is the subgroup size.
14291429
static std::unordered_map<std::string, std::unordered_map<std::string, uint32_t>> gpu_pipeline_config = {
1430-
{"RX 5700", {
1430+
{"NAVI1", {
14311431
{"soft_max_f32", 64}, {"soft_max_f32_wg512", 64},
14321432
{"soft_max_f32_f16", 64}, {"soft_max_f32_f16_wg512", 64},
14331433
{"im2col_f32", 64}, {"im2col_f32_f16", 64},
@@ -1584,7 +1584,7 @@ static void ggml_vk_load_shaders(vk_device& device) {
15841584

15851585
required_subgroup_size = get_subgroup_size(name, device_name);
15861586
if (required_subgroup_size == 0) {
1587-
required_subgroup_size = (device_name.find("RX 5700") != std::string::npos) ? 32 : required_subgroup_size;
1587+
required_subgroup_size = (device_name.find("NAVI1") != std::string::npos) ? 32 : required_subgroup_size;
15881588
}
15891589

15901590
if (!pipeline) {

0 commit comments

Comments
 (0)