Skip to content

Commit

Permalink
Update MetalExt.jl
Browse files Browse the repository at this point in the history
  • Loading branch information
matthieugomez committed Nov 19, 2023
1 parent 292b73c commit bc6345d
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion ext/MetalExt.jl
Original file line number Diff line number Diff line change
Expand Up @@ -119,7 +119,7 @@ end

function scale!(scale::MtlVector, refs::MtlVector, interaction::MtlVector, weights::MtlVector, nthreads::Integer)
nblocks = cld(length(refs), nthreads)
fill!(scale, 0.0)
fill!(scale, 0)
Metal.@sync @metal threads=nthreads groups=nblocks scale_kernel!(scale, refs, interaction, weights)
Metal.@sync @metal threads=nthreads groups=nblocks inv_kernel!(scale)
end
Expand Down

0 comments on commit bc6345d

Please sign in to comment.