metal : minor

This commit is contained in:
Georgi Gerganov 2023-09-07 15:33:21 +03:00
parent efac2d469f
commit 2f689dee06
No known key found for this signature in database
GPG Key ID: 449E073F9DC10735

View File

@ -111,7 +111,7 @@ kernel void kernel_soft_max(
uint3 tgpig[[threadgroup_position_in_grid]], uint3 tgpig[[threadgroup_position_in_grid]],
uint3 tpitg[[thread_position_in_threadgroup]], uint3 tpitg[[thread_position_in_threadgroup]],
uint3 ntg[[threads_per_threadgroup]]) { uint3 ntg[[threads_per_threadgroup]]) {
const int64_t i03 = tgpig[2]; const int64_t i03 = tgpig[2];
const int64_t i02 = tgpig[1]; const int64_t i02 = tgpig[1];
const int64_t i01 = tgpig[0]; const int64_t i01 = tgpig[0];