提交 9f448aa0 编写于 作者: L liuqi

Fix some bugs.

1. Add transform for Shape and InferConv2dShape ops.
2. Remove matmul gpu kernel.
3. Fix the length of abbreviated commit id in MACE version.
上级 61de4dab
...@@ -50,15 +50,6 @@ std::string TransformedName(const std::string &input_name, ...@@ -50,15 +50,6 @@ std::string TransformedName(const std::string &input_name,
return ss.str(); return ss.str();
} }
#ifdef MACE_ENABLE_OPENCL
bool TransformRequiredOp(const std::string &op_type) {
static const std::unordered_set<std::string> kNoTransformOp = {
"Shape", "InferConv2dShape"
};
return kNoTransformOp.count(op_type) == 0;
}
#endif // MACE_ENABLE_OPENCL
void BuildTransposeOpDef( void BuildTransposeOpDef(
const std::string &input_name, const std::string &input_name,
const std::string &output_name, const std::string &output_name,
...@@ -514,8 +505,6 @@ MaceStatus NetDefAdapter::AdaptMemoryType( ...@@ -514,8 +505,6 @@ MaceStatus NetDefAdapter::AdaptMemoryType(
// (only support one kind of memory type for multiple outputs) // (only support one kind of memory type for multiple outputs)
op_registry_->GetInOutMemoryTypes(op_def->type(), context); op_registry_->GetInOutMemoryTypes(op_def->type(), context);
#ifdef MACE_ENABLE_OPENCL #ifdef MACE_ENABLE_OPENCL
// if op is memory-unused op, no transformation
if (TransformRequiredOp(op_def->type())) {
int input_size = op_def->input_size(); int input_size = op_def->input_size();
for (int i = 0; i < input_size; ++i) { for (int i = 0; i < input_size; ++i) {
if (output_map->count(op_def->input(i)) == 0) { if (output_map->count(op_def->input(i)) == 0) {
...@@ -585,7 +574,6 @@ MaceStatus NetDefAdapter::AdaptMemoryType( ...@@ -585,7 +574,6 @@ MaceStatus NetDefAdapter::AdaptMemoryType(
op_def->set_input(i, transformed_name); op_def->set_input(i, transformed_name);
} }
} }
}
#else #else
MACE_UNUSED(output_map); MACE_UNUSED(output_map);
MACE_UNUSED(tensor_shape_map); MACE_UNUSED(tensor_shape_map);
......
...@@ -602,14 +602,6 @@ void RegisterMatMul(OpRegistryBase *op_registry) { ...@@ -602,14 +602,6 @@ void RegisterMatMul(OpRegistryBase *op_registry) {
DeviceType::CPU, uint8_t); DeviceType::CPU, uint8_t);
#endif // MACE_ENABLE_QUANTIZE #endif // MACE_ENABLE_QUANTIZE
#ifdef MACE_ENABLE_OPENCL
MACE_REGISTER_OP(op_registry, "MatMul", MatMulOp,
DeviceType::GPU, float);
MACE_REGISTER_OP(op_registry, "MatMul", MatMulOp,
DeviceType::GPU, half);
#endif // MACE_ENABLE_OPENCL
#if defined(MACE_ENABLE_NEON) && defined(__ANDROID__) #if defined(MACE_ENABLE_NEON) && defined(__ANDROID__)
MACE_REGISTER_OP(op_registry, "MatMul", MatMulOp, MACE_REGISTER_OP(op_registry, "MatMul", MatMulOp,
DeviceType::CPU, float16_t); DeviceType::CPU, float16_t);
......
...@@ -28,7 +28,7 @@ fi ...@@ -28,7 +28,7 @@ fi
mkdir -p $OUTPUT_DIR mkdir -p $OUTPUT_DIR
GIT_VERSION=$(git --git-dir=${MACE_SOURCE_DIR}/.git --work-tree=${MACE_SOURCE_DIR} describe --long --tags) GIT_VERSION=$(git --git-dir=${MACE_SOURCE_DIR}/.git --work-tree=${MACE_SOURCE_DIR} describe --long --tags --abbrev=7)
if [[ $? != 0 ]]; then if [[ $? != 0 ]]; then
GIT_VERSION=unknown GIT_VERSION=unknown
else else
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册