-
Notifications
You must be signed in to change notification settings - Fork 19
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Revised run command for CUDA testing #29
Comments
Also, in run_benchmark.py around line 126, you need to replace:
with
on a CUDA machine, the current code to empty the cache fails since there's no mps version of empty_cache on the cuda version of pytorch. |
Thanks @DaveSprague for reporting this issue, it was indeed an important error! |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Under the Run on other devices section, you have the command as:
python run_benchmark.py --include_mps=False --include_mlx_gpu=False --include_mlx_cpu=False --include_cuda=True --include_cpu=True
but it also needs to turn off the --include_mlx_gpu_compile option, so the correct command would be:
python run_benchmark.py --include_mps=False --include_mlx_gpu=False --include_mlx_gpu_compile=False --include_mlx_cpu=False --include_cuda=True --include_cpu=True
The text was updated successfully, but these errors were encountered: