DirectML is a high-performance, hardware-accelerated DirectX 12 library for machine learning. DirectML provides GPU acceleration for common machine learning tasks across a broad range of supported hardware and drivers, including all DirectX 12-capable GPUs from vendors such as AMD, Intel, NVIDIA, and Qualcomm.
MIT License
2.24k
stars
299
forks
source link
'aten::count_nonzero.dim_intlist' is not currently supported on the dml backend and will fall back to run on the cpu' #665
I'm trying to use ComfyUI running SDXL with a AMD GPU, after clicking to generate the prompt i get this:
userwarning: the operator 'aten::count_nonzero.dim_intlist' is not currently supported on the dml backend and will fall back to run on the cpu. this may have performance implications. (triggered internally at c:__w\1\s\pytorch-directml-plugin\torch_directml\csrc\dml\dml_cpu_fallback.cpp:17.) if latent_image is not none and torch.count_nonzero(latent_image) > 0: #don't shift the empty latent image.
I'm trying to use ComfyUI running SDXL with a AMD GPU, after clicking to generate the prompt i get this:
userwarning: the operator 'aten::count_nonzero.dim_intlist' is not currently supported on the dml backend and will fall back to run on the cpu. this may have performance implications. (triggered internally at c:__w\1\s\pytorch-directml-plugin\torch_directml\csrc\dml\dml_cpu_fallback.cpp:17.) if latent_image is not none and torch.count_nonzero(latent_image) > 0: #don't shift the empty latent image.