Fix nvidia with Catleap support