スポンサーリンク

CUDAのベクトル型を加減乗除する演算子

CUDAにはfloat3型などのベクトル型はあるが、ベクトル同士を計算する+,-などの演算子は定義されていない。

ではどうするかというと、CUDAのサンプルの中にhelper_math.hがあるので、これをダウンロードしてプロジェクトに追加する。

ダウンロード

https://github.com/NVIDIA/cuda-samples

導入

cuda-samples-master/Common/helper_math.h

使用例

#include "helper_math.h"

// helper_math.h でincludeされているので不要
// #include "cuda_runtime.h"

#include "device_launch_parameters.h"

float3 my_calc(float3 a, float3 b) {
  return a + b;
}

コメントを残す

メールアドレスが公開されることはありません。

日本語が含まれない投稿は無視されますのでご注意ください。(スパム対策)


この記事のトラックバックURL: