zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-09-06 08:24:39 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

[CUDA] synch properly waits for all tasks to finish and clear (#2303)

Browse Source 
* cuda synch properly waits for all tasks to finish and clear

* fix copy
This commit is contained in:
Awni Hannun
2025-06-17 12:03:25 -07:00
committed by GitHub
parent b8022c578a
commit cad5c0241c
7 changed files with 27 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
python/tests/cuda_skip.py
View File

@@ -6,7 +6,6 @@ cuda_skip = {
"TestEinsum.test_ellipses",
"TestEinsum.test_opt_einsum_test_cases",
"TestLoad.test_load_f8_e4m3",
"TestMemory.test_memory_info",
"TestLayers.test_group_norm",
"TestLayers.test_pooling",
"TestLayers.test_quantized_embedding",