[BUG]when exporting hugeCTR model, NVT model is hardcoded to run on CPU

Open mengdong opened this issue 4 years ago • 2 comments

Describe the bug see https://github.com/NVIDIA-Merlin/NVTabular/blob/72f2fe4e882e126856dd96054807acbc929f64e3/nvtabular/inference/triton/init.py#L427-L428

Steps/Code to reproduce bug NA

Expected behavior NVT backend should be allowed to run on GPU

Environment details (please complete the following information): NA

Oct 29 '21 18:10 mengdong

@bschifferer can you confirm this is still the case?

Jun 10 '22 00:06 EvenOldridge

As far as I can tell, this still appears to be the case

Jun 13 '22 18:06 karlhigley