张壹 zhangyiss
0 Followers · 0 Following
  • Joined on 2024-09-10
Repositories
54
 Projects Packages Public Activity Starred Repositories
zhangyiss synced commits to refs/pull/2476/merge at zhangyiss/mlx from mirror 2025-08-09 10:46:40 +08:00
ab1baf722b Merge 984cefb14d into 7fde1b6a1e
984cefb14d CUDA_VISIBLE_DEVICES to local rank
7fde1b6a1e Fix logsumexp/softmax not fused for some cases (#2474)
Compare 3 commits »
zhangyiss synced commits to refs/pull/2477/merge at zhangyiss/mlx from mirror 2025-08-09 10:46:40 +08:00
d91f5ae900 Merge c28cff7763 into 7fde1b6a1e
7fde1b6a1e Fix logsumexp/softmax not fused for some cases (#2474)
Compare 2 commits »
zhangyiss synced and deleted reference refs/tags/refs/pull/2474/merge at zhangyiss/mlx from mirror 2025-08-09 10:46:39 +08:00
zhangyiss synced commits to main at zhangyiss/mlx from mirror 2025-08-09 10:46:39 +08:00
7fde1b6a1e Fix logsumexp/softmax not fused for some cases (#2474)
zhangyiss synced commits to refs/pull/2472/merge at zhangyiss/mlx from mirror 2025-08-09 10:46:39 +08:00
7a09320b3a Merge 7ebb9fe6683b3a431edd2e098bce77bbfd49ab5e into 7fde1b6a1e
7fde1b6a1e Fix logsumexp/softmax not fused for some cases (#2474)
Compare 2 commits »
zhangyiss synced commits to refs/pull/2476/head at zhangyiss/mlx from mirror 2025-08-09 10:46:39 +08:00
984cefb14d CUDA_VISIBLE_DEVICES to local rank
zhangyiss synced commits to refs/pull/2156/merge at zhangyiss/mlx from mirror 2025-08-09 02:36:43 +08:00
f0e43435d0 Merge 7c99acb799 into aa7b47481a
aa7b47481a [CUDA] Optimize set_mm_device_pointers for small ndim (#2473)
Compare 2 commits »
zhangyiss synced commits to refs/pull/2300/merge at zhangyiss/mlx from mirror 2025-08-09 02:36:43 +08:00
be1ba9da6a Merge cc4de6a607 into aa7b47481a
aa7b47481a [CUDA] Optimize set_mm_device_pointers for small ndim (#2473)
Compare 2 commits »
zhangyiss synced commits to refs/pull/2401/merge at zhangyiss/mlx from mirror 2025-08-09 02:36:43 +08:00
2be596335e Merge c7cdd51f50 into aa7b47481a
aa7b47481a [CUDA] Optimize set_mm_device_pointers for small ndim (#2473)
Compare 2 commits »
zhangyiss synced commits to refs/pull/1970/merge at zhangyiss/mlx from mirror 2025-08-09 02:36:42 +08:00
4db830e256 Merge 688e421184 into aa7b47481a
aa7b47481a [CUDA] Optimize set_mm_device_pointers for small ndim (#2473)
Compare 2 commits »
zhangyiss synced commits to refs/pull/2074/merge at zhangyiss/mlx from mirror 2025-08-09 02:36:42 +08:00
20b098dce1 Merge eeaf1fa463 into aa7b47481a
aa7b47481a [CUDA] Optimize set_mm_device_pointers for small ndim (#2473)
Compare 2 commits »
zhangyiss synced commits to refs/pull/2474/merge at zhangyiss/mlx from mirror 2025-08-08 18:16:45 +08:00
94c4b88ac3 Merge 79f46f1146 into aa7b47481a
79f46f1146 Fix logsumexp/softmax not fused for some cases
aa7b47481a [CUDA] Optimize set_mm_device_pointers for small ndim (#2473)
Compare 3 commits »
zhangyiss synced commits to refs/pull/2476/merge at zhangyiss/mlx from mirror 2025-08-08 18:16:45 +08:00
1c7ce7856c Merge dadf8d9c93 into aa7b47481a
aa7b47481a [CUDA] Optimize set_mm_device_pointers for small ndim (#2473)
Compare 2 commits »
zhangyiss synced commits to refs/pull/2472/merge at zhangyiss/mlx from mirror 2025-08-08 18:16:44 +08:00
76447a3dff Merge 7ebb9fe6683b3a431edd2e098bce77bbfd49ab5e into aa7b47481a
aa7b47481a [CUDA] Optimize set_mm_device_pointers for small ndim (#2473)
Compare 2 commits »
zhangyiss synced commits to refs/pull/2474/head at zhangyiss/mlx from mirror 2025-08-08 18:16:44 +08:00
79f46f1146 Fix logsumexp/softmax not fused for some cases
56be773610 version (#2470)
Compare 2 commits »
zhangyiss synced commits to refs/pull/2104/merge at zhangyiss/mlx from mirror 2025-08-08 18:16:43 +08:00
5016b676be Merge 992eac905a into aa7b47481a
aa7b47481a [CUDA] Optimize set_mm_device_pointers for small ndim (#2473)
Compare 2 commits »
zhangyiss synced commits to refs/pull/2290/merge at zhangyiss/mlx from mirror 2025-08-08 18:16:43 +08:00
0490c2a329 Merge cb4dc59a9e into aa7b47481a
aa7b47481a [CUDA] Optimize set_mm_device_pointers for small ndim (#2473)
Compare 2 commits »
zhangyiss synced and deleted reference refs/tags/refs/pull/2473/merge at zhangyiss/mlx from mirror 2025-08-08 18:16:42 +08:00
zhangyiss synced commits to main at zhangyiss/mlx from mirror 2025-08-08 18:16:42 +08:00
aa7b47481a [CUDA] Optimize set_mm_device_pointers for small ndim (#2473)
zhangyiss synced commits to refs/pull/2401/merge at zhangyiss/mlx from mirror 2025-08-08 09:56:48 +08:00
ba2209d465 Merge c7cdd51f50 into 56be773610
56be773610 version (#2470)
a9bdd67baa Add CUDA sdpa vector (#2468)
f2adb5638d Fix typo in metal command encoder (#2471)
728d4db582 Support destination arg in tree flatten/unflatten (#2450)
Compare 7 commits »