You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
# Mbsize, Mrem, Mremfinal, _to_spawn = split_m(M, tospawn, mᵣW) # M is guaranteed to be > W because of `W ≥ M` condition for `jmultsplitn!`...
361
359
Mblock_Mrem, Mblock_ =promote(Mbsize + W, Mbsize)
362
360
u_to_spawn = _to_spawn % UInt
363
361
tid =0
@@ -366,11 +364,11 @@ function matmul_pack_A_and_B!(
366
364
last_id = _to_spawn -One()
367
365
for m ∈CloseOpen(last_id) # ...thus the fact that `CloseOpen()` iterates at least once is okay.
368
366
Mblock =ifelse(m < Mrem, Mblock_Mrem, Mblock_)
369
-
launch_thread_mul!(C, A, B, α, β, Mblock, K, N, p, bc_ptr, m % UInt, u_to_spawn, StaticFloat{W₁}(),StaticFloat{W₂}(),StaticFloat{R₁}(),StaticFloat{R₂}())
367
+
launch_thread_mul!(C, A, B, α, β, Mblock, K, N, p, bc_ptr, m % UInt, u_to_spawn, StaticFloat64{W₁}(),StaticFloat64{W₂}(),StaticFloat64{R₁}(),StaticFloat64{R₂}())
370
368
A =gesp(A, (Mblock, Zero()))
371
369
C =gesp(C, (Mblock, Zero()))
372
370
end
373
-
sync_mul!(C, A, B, α, β, Mremfinal, K, N, p, bc_ptr, last_id % UInt, u_to_spawn, StaticFloat{W₁}(), StaticFloat{W₂}(), StaticFloat{R₁}(), StaticFloat{R₂}())
371
+
sync_mul!(C, A, B, α, β, Mremfinal, K, N, p, bc_ptr, last_id % UInt, u_to_spawn, StaticFloat64{W₁}(), StaticFloat64{W₂}(), StaticFloat64{R₁}(), StaticFloat64{R₂}())
374
372
waitonmultasks(CloseOpen(One(), _to_spawn))
375
373
end
376
374
_free_bcache!(bc)
@@ -379,11 +377,11 @@ end
379
377
380
378
functionsync_mul!(
381
379
C::AbstractStridedPointer{T}, A::AbstractStridedPointer, B::AbstractStridedPointer, α, β, M, K, N, atomicp::Ptr{UInt}, bc::Ptr, id::UInt, total_ids::UInt,
0 commit comments