Browse Source

Merge branch 'master' of git+ssh://scm.gforge.inria.fr/gitroot/starpu/starpu into ft_checkpoint

Romain LION 5 years ago
parent
commit
aa6cf4f5f7
2 changed files with 1 additions and 5 deletions
  1. 0 1
      julia/examples/gemm/gemm.jl
  2. 1 4
      julia/src/compiler/file_generation.jl

+ 0 - 1
julia/examples/gemm/gemm.jl

@@ -15,7 +15,6 @@
 #
 using StarPU
 using LinearAlgebra.BLAS
-using BenchmarkTools
 
 @target STARPU_CPU+STARPU_CUDA
 @codelet function gemm(A :: Matrix{Float32}, B :: Matrix{Float32}, C :: Matrix{Float32}, alpha :: Float32, beta :: Float32) :: Nothing

+ 1 - 4
julia/src/compiler/file_generation.jl

@@ -115,10 +115,6 @@ macro codelet(x)
     c_struct_param_decl = generate_c_struct_param_declaration(name)
     cpu_expr = transform_to_cpu_kernel(parsed)
 
-    if (starpu_target & STARPU_CUDA != 0)
-        prekernel, kernel = transform_to_cuda_kernel(parsed)
-    end
-
     generated_cpu_kernel_file_name=string("genc_",string(x.args[1].args[1].args[1]),".c")
     generated_cuda_kernel_file_name=string("gencuda_",string(x.args[1].args[1].args[1]),".cu")
 
@@ -136,6 +132,7 @@ macro codelet(x)
         kernel_file = open(generated_cuda_kernel_file_name, "w")
         debug_print("generating ", generated_cuda_kernel_file_name)
         print(kernel_file, cuda_kernel_file_start)
+        prekernel, kernel = transform_to_cuda_kernel(parsed)
 
         if kernel != nothing
             print(kernel_file, "__global__ ", kernel)