Selaa lähdekoodia

julia: Fix cublas.

Pierre Huchant 5 vuotta sitten
vanhempi
commit
1784fb9ff5
1 muutettua tiedostoa jossa 2 lisäystä ja 2 poistoa
  1. 2 2
      julia/src/compiler/cuda.jl

+ 2 - 2
julia/src/compiler/cuda.jl

@@ -153,8 +153,8 @@ function translate_cublas(expr :: StarpuExpr)
                               :STARPU_DGEMV  => (:cublasDgemv, [1], [], [4,9], [], []),
                               :STARPU_SSCAL  => (:cublasSscal, [], [], [2], [], []),
                               :STARPU_DSCAL  => (:cublasDscal, [], [], [2], [], []),
-                              :STARPU_STRSM  => (:cublasStrsm, [3], [2], [7,10], [1], [4]),
-                              :STARPU_DTRSM  => (:cublasDtrsm, [3], [2], [7,10], [1], [4]),
+                              :STARPU_STRSM  => (:cublasStrsm, [3], [2], [7], [1], [4]),
+                              :STARPU_DTRSM  => (:cublasDtrsm, [3], [2], [7], [1], [4]),
                               :STARPU_SSYR   => (:cublasSsyr, [], [1], [3], [], []),
                               :STARPU_SSYRK  => (:cublasSsyrk, [2], [1], [5,8], [], []),
                               :STARPU_SGER   => (:cublasSger, [], [], [3], [], []),