Antares: an automatic engine for multi-platform kernel generation and optimization. Supporting CPU, CUDA, ROCm, DirectX12, GraphCore, SYCL for CPU/GPU, OpenCL for AMD/NVIDIA, Android CPU/GPU backends.
Other
435
stars
45
forks
source link
Can antares assign specified gpus for evaluation? #358
I have four gpus in my environment, each time I evaluated cuda backend, all the four gpus will run evaluator program, but I just wanna antare uses no more than two of them because I also need another gpus to profile other programs.
so can antares assign specified gpus for evaluation, actually I tried setting CUDA_VISIBLE_DEVICES but it didn't work, as I expected...
I have four gpus in my environment, each time I evaluated cuda backend, all the four gpus will run evaluator program, but I just wanna antare uses no more than two of them because I also need another gpus to profile other programs.
so can antares assign specified gpus for evaluation, actually I tried setting
CUDA_VISIBLE_DEVICES
but it didn't work, as I expected...