aboutsummaryrefslogtreecommitdiff
path: root/test/Concat.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'test/Concat.cpp')
-rw-r--r--test/Concat.cpp2
1 files changed, 1 insertions, 1 deletions
diff --git a/test/Concat.cpp b/test/Concat.cpp
index 02d66cb8..9beb67bd 100644
--- a/test/Concat.cpp
+++ b/test/Concat.cpp
@@ -124,7 +124,7 @@ ConcatTestImpl(const std::vector<const TestTensor*> & inputs,
}
// add memory for the output
- android::sp<IMemory> outMemory = AddPoolAndGetData(expectedOutputTensor.GetNumElements(), request);
+ android::sp<IMemory> outMemory = AddPoolAndGetData<float>(expectedOutputTensor.GetNumElements(), request);
float* outdata = static_cast<float*>(static_cast<void*>(outMemory->getPointer()));
// run the execution