search

microsoft / antares

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

Closed LeiWang1999 closed 2 years ago

LeiWang1999 commented 2 years ago

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...

ghostplant commented 2 years ago

DEVICE_ID=3 <.. other configs ..> antares