通过使用Aparapi进行显式缓冲管理,我遇到了问题。
以下代码显示我正在尝试管理多个put / get循环以从GPU刷新/获取数据。似乎第一个put
和get
已完成,但其他人未完成。
import com.amd.aparapi
// Dummy test to reproduce explicit buffer management
public class QuickTestExplicit extends Kernel
{
private static final float DELTA = (float) 1E-5;
// will be filled, put on GPU in each iterations
private float[][] values;
// will be filled with results, put in GPU once but retrieved several times
private float[] currentRes;
private void initData()
{
values = new float[2000][20];
currentRes = new float[2000];
}
@Override
public void run()
{
int id = getGlobalId();
long accum = 0;
// simple sum of elements
for (int index = 0; index < 20; ++index)
{
accum += values[id][index];
}
currentRes[id] = accum;
}
public void process()
{
boolean passed = true;
initData();
if (isExplicit())
{
put(currentRes);
}
for (int row = 0; row < 2000; ++row)
{
for (int i = 0; i < values.length; ++i)
{
for (int depth = 0; depth < 20; ++depth)
{
values[i][depth] = (float) row;
}
}
if (isExplicit())
{
put(values);
}
execute(values.length);
if (isExplicit())
{
get(currentRes);
}
// just check the success of the operation (for the example)
passed = true;
for (int currentIndexRes = 0; currentIndexRes < currentRes.length; ++currentIndexRes)
{
passed &= Math.abs(currentRes[currentIndexRes] - (row * 20.0)) < DELTA;
}
if (passed)
{
System.out.println("ROW " + row + " PASSED");
}
else
{
System.out.println("ROW " + row + " FAILED");
}
}
}
public static void main(String[] args)
{
QuickTestExplicit kern = new QuickTestExplicit();
kern.setExecutionMode(EXECUTION_MODE.GPU);
kern.setExplicit(true);
kern.process();
}
}
所以我的问题是:
我不认为这是一个与GPU内存容量相关的问题(在我的情况下是2GB内存,应用程序只是放了2000 * 20 * 4 + 2000 * 4 = 168KB) 我使用的是CUDA架构。 仅供参考,此程序在以JTP模式运行时通过。
提前致谢!
编辑:我忘了提到我正在使用&#34; Aparapi_2014_04_29_Linux64&#34;版本可在svn / trunk / Downloads&#34;。中使用 似乎在使用2D Java原始数组时出现问题。我通过使用一维Java原始数组重写了相同的算法,并且完美地工作......任何想法?