C++ AMP中GPU并行计算

最新推荐文章于 2024-08-27 16:36:12 发布

Augusdi

最新推荐文章于 2024-08-27 16:36:12 发布

阅读量5.3k

点赞数

分类专栏： C++ AMP

C++ AMP 专栏收录该内容

75 篇文章 2 订阅

订阅专栏

AMP(Gpu并行计算，c#，vc++11) 学习（一）

感觉AMP代码很好懂

一、vc++11代码

#include <amp.h>
  
using namespace concurrency;
 
extern "C" __declspec ( dllexport ) void _stdcall square_array(float* arr, int n)
{
    // Create a view over the data on the CPU
    array_view<float,1> dataView(n, &arr[0]);
  
    // Run code on the GPU
    parallel_for_each(dataView.extent, [=] (index<1> idx) restrict(amp)
    {
        dataView[idx] = dataView[idx] * dataView[idx];
    });
  
    // Copy data from GPU to CPU
    dataView.synchronize();
}

Concurrency是和C#的线程安全数据结构名称一样的。所以很容易理解

parallel_for_each 类似 C#的 Parallel.ForEach

第12行的lambda 更是c#里面天天用的，看到以后很激动

dataView.extend 貌似是需要同时启动的算子数目

[=] 是 [dataView] 的缩写，

在c#里面lambda是没有“[]”这个的， []表示要进入lambda函数闭包内的变量。

c#是自动分析的，而vc++11需要手工列出传入的变量。

array_view（AMP引用数组）传值

array 传引用（AMP数组）（传引用加&）

restrict(amp) 是限定这个函数进行语法检查的一个标记，括号里可以写amp 或cpu

如果是amp （GPU）则，函数体里面的内容只能是c++11的一个子集，有些写法是不允许的。具体msdn上有描述

函数体不解释。至于Idx可以理解为并行算子的编号。

dataView.synchronize();

将GPU上的数据写回CPU

很简单真的，几乎没有一行不明白。

===============================================================================

二、c#调用代码

[DllImport("Win32Project1", CallingConvention = CallingConvention.StdCall)]
extern unsafe static void square_array(float* array, int length);
 
unsafe void Do()
{
    // Allocate an array
    float[] arr = this.textBox1.Text
        .Split(' ')
        .Select(v=> Convert.ToSingle(v) ).ToArray();
 
    // Square the array elements using C++ AMP
    fixed (float* arrPt = &arr[0])
    {
        square_array(arrPt, arr.Length);
    }
 
 
    this.textBox2.Text = string.Join(" ", arr);
}

读取textbox的字符串转换为浮点数组，然后调用amp函数，然后输出到另外一个textBox中。