fixup! Add device/random.jl

JuliaGPU · Mar 29, 2023 · 2439b9d · 2439b9d
1 parent 3cbaab3
commit 2439b9d
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/src/device/random.jl b/src/device/random.jl
@@ -66,8 +66,8 @@ end
 
 @inline function Base.getproperty(rng::Philox2x32, field::Symbol)
     threadId = workitemIdx().x + (workitemIdx().y - Int32(1)) * workgroupDim().x +
-                               (workitemIdx().z - Int32(1)) * workgroupDim().x * workgroupDim().y
-    warpId = (threadId - Int32(1)) >> 0x5 + Int32(1)  # fld1
+                                 (workitemIdx().z - Int32(1)) * workgroupDim().x * workgroupDim().y
+    warpId = (threadId - Int32(1)) >> 0x6 + Int32(1)  # fld1 by 64
 
     if field === :seed
         @inbounds global_random_seed()[1]
@@ -85,8 +85,8 @@ end
 
 @inline function Base.setproperty!(rng::Philox2x32, field::Symbol, x)
     threadId = workitemIdx().x + (workitemIdx().y - Int32(1)) * workgroupDim().x +
-                               (workitemIdx().z - Int32(1)) * workgroupDim().x * workgroupDim().y
-    warpId = (threadId - Int32(1)) >> 0x5 + Int32(1)  # fld1
+                                 (workitemIdx().z - Int32(1)) * workgroupDim().x * workgroupDim().y
+    warpId = (threadId - Int32(1)) >> 0x6 + Int32(1)  # fld1 by 64
 
     if field === :key
         @inbounds global_random_keys()[warpId] = x