Привет, мир с устройства + PYCUDA

Question

Я начинаю изучать PyCUDA в Google Colab. Я пытаюсь запустить пример "printf".

Все работает нормально, но я не получаю вывод на последней строке. Как я могу решить это?

import pycuda.driver as drv
import pycuda.autoinit
from pycuda.compiler import SourceModule

mod = SourceModule("""
#include <stdio.h>

  __global__ void myfirst_kernel()
  {
    printf("Hello,PyCUDA!!!");
  }
""")

function = mod.get_function("myfirst_kernel")
function(block=(4,4,1)) 

# Flush context printf buffer 
cuda.Context.synchronize()