Merge pull request #126 from frasercrmck/sub-group-size-1

[mux] Advertise a sub-group size of 1
uxlfoundation · Sep 12, 2023 · 0260608 · 0260608
2 parents 3a05571 + aba08ee
commit 0260608
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 0 deletions.
diff --git a/modules/mux/targets/host/source/device.cpp b/modules/mux/targets/host/source/device.cpp
@@ -523,6 +523,12 @@ device_info_s::device_info_s(host::arch arch, host::os os, bool native,
   this->sub_groups_support_ifp = false;
   this->supports_work_group_collectives = true;
   this->supports_generic_address_space = true;
+
+  static std::array<size_t, 1> sg_sizes = {
+      1,  // we can always produce a 'trivial' sub-group if asked.
+  };
+  this->sub_group_sizes = sg_sizes.data();
+  this->num_sub_group_sizes = sg_sizes.size();
 }
 
 host::arch device_info_s::detectHostArch() {

diff --git a/modules/mux/targets/riscv/source/device_info.cpp b/modules/mux/targets/riscv/source/device_info.cpp
@@ -145,6 +145,12 @@ device_info_s::device_info_s()
   this->max_hardware_counters = std::numeric_limits<int>::max();
   this->supports_work_group_collectives = true;
   this->supports_generic_address_space = true;
+
+  static std::array<size_t, 1> sg_sizes = {
+      1,  // we can always produce a 'trivial' sub-group if asked.
+  };
+  this->sub_group_sizes = sg_sizes.data();
+  this->num_sub_group_sizes = sg_sizes.size();
 }
 
 mux_result_t GetDeviceInfos(uint32_t device_types, uint64_t device_infos_length,