lhutton1 commented on a change in pull request #6109:
URL: https://github.com/apache/incubator-tvm/pull/6109#discussion_r459604438
##########
File path: src/runtime/contrib/arm_compute_lib/acl_utils.cc
##########
@@ -108,6 +111,30 @@ arm_compute::PadStrideInfo ToACLPadStride(const
std::vector<std::string>& pad,
arm_compute::DimensionRoundingType::FLOOR);
}
+arm_compute::DataType MakeDataType(const DLDataType& data_type) {
+ if (data_type.code == DLDataTypeCode::kDLFloat && data_type.bits == 32) {
+ return arm_compute::DataType::F32;
+ } else if (data_type.code == DLDataTypeCode::kDLUInt && data_type.bits == 8)
{
+ return arm_compute::DataType::QASYMM8;
+ } else if (data_type.code == DLDataTypeCode::kDLInt && data_type.bits == 32)
{
+ return arm_compute::DataType::S32;
+ } else {
+ LOG(FATAL) << "Datatype " << data_type << " unsupported by ACL runtime";
+ return arm_compute::DataType::UNKNOWN;
+ }
+}
+
+template <typename T>
+std::vector<T> GetVectorFromDLTensor(const DLTensor* tensor) {
Review comment:
I had a look for something like this in the runtime, although couldn't
find anything yet
----------------------------------------------------------------
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
For queries about this service, please contact Infrastructure at:
[email protected]
