torch.utils.hipify.hipify_python.hip_header_magic

torch.utils.hipify.hipify_python.hip_header_magic(input_string) [исходный код]

Если файл содержит вызовы встроенных функций ядра и не включает заголовок cuda_runtime.h, автоматически добавляется директива #include, соответствующая «магическим» подключениям, предоставляемым NVCC. .. todo:: Обновить логику, чтобы игнорировать случаи, когда cuda_runtime.h включается другим файлом.