Я хочу запустить код Halide Jited на нескольких платформах GPU.Я должен определить, какое устройство использовать.Обычный способ использовать пользовательское устройство - заменить api halide_cuda_acquire_context / halide_cuda_release_context, это работает для режима компиляции AOT.Но это не работает для кода Jited в Halide.Как заменить эти API в Jited-коде?