include("../src/Compiler/include.jl")

starpu_new_cpu_kernel_file("build/generated_cpu_mult.c")
starpu_new_cuda_kernel_file("build/generated_cuda_mult.cu")

@cpu_cuda_kernel function matrix_mult(m1 :: Matrix{Float32}, m2 :: Matrix{Float32}, m3 :: Matrix{Float32}) :: Void

    width_m2 :: Int64 = width(m2)
    height_m1 :: Int64 = height(m1)
    width_m1 :: Int64 = width(m1)

    @indep for j in (1 : width_m2)
        @indep for i in (1 : height_m1)

            sum :: Float32 = 0.

            for k in (1 : width_m1)
                sum = sum + m1[i, k] * m2[k, j]
            end

            m3[i, j] = sum
        end
    end
end

compile_cpu_kernels("build/generated_cpu_mult.so")
compile_cuda_kernels("build/generated_cuda_mult.so")
combine_kernel_files("build/generated_tasks.so", ["build/generated_cpu_mult.so", "build/generated_cuda_mult.so"])