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

现在的位置: 首页 > 综合 > 正文

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

2012年06月04日 ⁄ 综合 ⁄ 共 1616字 ⁄ 字号小中大 ⁄ 评论关闭

感觉AMP代码很好懂

一、vc++11代码

#include "stdafx.h"
#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);
}