我使用以下代码将浮点数数组传输到renderscript内核:
float[] bufName = new float[3];
bufName [0] = 255;
bufName [1] = 255;
bufName [2] = 0;
Allocation alloc1 = Allocation.createSized(mRs, Element.F32(mRs), 3);
alloc1.copy1DRangeFrom(0, 3, mtmd);
ScriptC_foo foo = new ScriptC_foo(mRs);
foo.set_gIn(alloc1);
我在
gIn
文件中定义了foo.rs
,如下所示:rs_allocation gIn;
我想使用16位浮点数。我知道我应该将分配创建更改为此:
Allocation alloc1 = Allocation.createSized(mRs, Element.F16(mRs), 3);
但是,我找不到将bufName数组复制到分配的解决方案。任何帮助表示赞赏。
最佳答案
Java没有定义半精度浮点数,因此您必须自己进行操作才能使其工作。如果使用Float.valueOf(f).shortValue()
(其中f
是您要表示为半精度的特定流)。这应该正确地将float
降低为新的位大小。然后创建Allocation
大小的Element.F16
。您应该能够将short
值的数组复制到RenderScript中以完成工作。