more erros in circular :(

iitis · Aug 23, 2019 · 669b24f · 669b24f
1 parent 2e08c77
commit 669b24f
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 4 deletions.
diff --git a/curandommatrices/src/circular.jl b/curandommatrices/src/circular.jl
@@ -1,5 +1,6 @@
 # FIXME: this can be accelerated by writing a custom kernel
 function angle!(a)
+    a = CuArray(a)
     # @cuprintf("thread %ld", Int64(1))
     ℜ = real.(a)
     # @cuprintf("thread %ld", Int64(1))
@@ -10,9 +11,11 @@ function angle!(a)
     # @cuprintf("thread %ld", Int64(1))
     stride = blockDim().x * gridDim().x
     # @cuprintf("thread %ld", Int64(1))
-    for i=index:stride:length(ℜ)
-        a[i] = (ℑ[i], ℜ[i])
-    end
+    i = (blockIdx().x-1) * blockDim().x + threadIdx().x
+    a[i] = ℜ[i] + ℑ[i]
+    # for i=index:stride:length(ℜ)
+    #     a[i] = +(ℑ[i], ℜ[i])
+    # end
     return nothing
 end
 

diff --git a/curandommatrices/src/ginibre.jl b/curandommatrices/src/ginibre.jl
@@ -1,5 +1,5 @@
 curand(g::GinibreEnsemble{1}) = CuArrays.randn(g.m, g.n)
-curand(g::GinibreEnsemble{2}) = CuArrays.randn( g.m, g.n)+1im*CuArrays.randn(g.m, g.n)
+curand(g::GinibreEnsemble{2}) = CuArrays.randn(g.m, g.n)+1im*CuArrays.randn(g.m, g.n)
 
 function curand(g::GinibreEnsemble{4})
     q0=CuArrays.randn(g.m, g.n)