KI Sub-groups

[christiangnrd]

No description provided.

[github-actions]

Your PR requires formatting changes to meet the project's style guidelines.
Please consider running Runic (git runic main) to apply these changes.

Click here to view the suggested changes.

diff --git a/src/intrinsics.jl b/src/intrinsics.jl
index 5f5528c7..4884c87f 100644
--- a/src/intrinsics.jl
+++ b/src/intrinsics.jl
@@ -269,7 +269,7 @@ kernel on the host.
 
     Backends must also implement the on-device kernel launch functionality.
 """
-struct Kernel{B,Kern}
+struct Kernel{B, Kern}
     backend::B
     kern::Kern
 end
@@ -410,7 +410,7 @@ There are a few keyword arguments that influence the behavior of `KI.@kernel`:
 """
 macro kernel(backend, ex...)
     call = ex[end]
-    kwargs = map(ex[1:(end-1)]) do kwarg
+    kwargs = map(ex[1:(end - 1)]) do kwarg
         if kwarg isa Symbol
             :($kwarg = $kwarg)
         elseif Meta.isexpr(kwarg, :(=))
diff --git a/test/intrinsics.jl b/test/intrinsics.jl
index 8dcddfc7..076fd3e3 100644
--- a/test/intrinsics.jl
+++ b/test/intrinsics.jl
@@ -12,12 +12,14 @@ function test_intrinsics_kernel(results)
     i = KI.get_global_id().x
 
     if i <= length(results)
-        @inbounds results[i] = KernelData(KI.get_global_size().x,
-                                          KI.get_global_id().x,
-                                          KI.get_local_size().x,
-                                          KI.get_local_id().x,
-                                          KI.get_num_groups().x,
-                                          KI.get_group_id().x)
+        @inbounds results[i] = KernelData(
+            KI.get_global_size().x,
+            KI.get_global_id().x,
+            KI.get_local_size().x,
+            KI.get_local_id().x,
+            KI.get_num_groups().x,
+            KI.get_group_id().x
+        )
     end
     return
 end
@@ -32,11 +34,13 @@ function test_subgroup_kernel(results)
     i = KI.get_global_id().x
 
     if i <= length(results)
-        @inbounds results[i] = SubgroupData(KI.get_sub_group_size(),
-                                            KI.get_max_sub_group_size(),
-                                            KI.get_num_sub_groups(),
-                                            KI.get_sub_group_id(),
-                                            KI.get_sub_group_local_id())
+        @inbounds results[i] = SubgroupData(
+            KI.get_sub_group_size(),
+            KI.get_max_sub_group_size(),
+            KI.get_num_sub_groups(),
+            KI.get_sub_group_id(),
+            KI.get_sub_group_local_id()
+        )
     end
     return
 end
diff --git a/test/localmem.jl b/test/localmem.jl
index 7894ee61..c1d63c7a 100644
--- a/test/localmem.jl
+++ b/test/localmem.jl
@@ -57,7 +57,7 @@ end
     lmem1 = @localmem Int (N,) # Ok iff groupsize is static
     lmem2 = @localmem Int (N,) # Ok iff groupsize is static
     @inbounds begin
-        lmem1[i] = i-1
+        lmem1[i] = i - 1
         lmem2[i] = 1
         @synchronize
         A[I] = lmem1[N2 - i + 1] + lmem2[N2 - i + 1]