Google Colab で CUDA を正しく実行したい

Question

前提・実現したいこと

Google Colab で CUDA を実行したのですが、実行結果が想定と異なるものになってしまいます。
エラーは発生しませんでした。

発生している問題

想定している実行結果

CPU
GPU
GPU

実際の実行結果

CPU

ソースコード

#include <stdio.h>

__global__ void testFromGPU() {
    printf("GPU\n");
}

int main() {
    printf("CPU\n");
    testFromGPU<<<1, 2>>>();
    return 0;
}

実行するまでの手順

1: 「ランタイムのタイプの変更」からハードウェアアクセラレータを GPU に設定
2: NVCCを実行できるようにするために以下を実行

!pip install git+git://github.com/andreinechaev/nvcc4jupyter.git
%load_ext nvcc_plugin

3: 該当のソースコードの先頭に「%%cu」を挿入して実行

%%cu

#include <stdio.h>

__global__ void testFromGPU() {
    printf("GPU\n");
}

int main() {
    printf("CPU\n");
    testFromGPU<<<1, 2>>>();
    return 0;
}

Comments

[jbpb0]

これかな
https://qiita.com/JmpM/items/ada670ec80be9566269e

Answer 1

    testFromGPU<<<1, 2>>>();

↓ 同期処理を追加

    testFromGPU<<<1, 2>>>();
    cudaDeviceSynchronize();

参考
並列"Hello World"から始めるCUDA入門