提交 1a6f62bd 编写于 作者: V VectorSL

gpu update type check

上级 2f65bf43
......@@ -68,13 +68,18 @@ std::string GpuKernelFactory::SupportedTypeList(const std::string &kernel_name)
return type_lists;
}
for (size_t attr_index = 0; attr_index < (iter->second).size(); ++attr_index) {
std::string type_list = "[";
std::string type_list = "in[";
auto attr = (iter->second)[attr_index].first;
for (size_t input_index = 0; input_index < attr.GetInputSize(); ++input_index) {
type_list = type_list + TypeId2String(attr.GetInputAttr(input_index).first) +
((input_index == (attr.GetInputSize() - 1)) ? "" : " ");
}
type_lists = type_lists + type_list + "] ";
type_list = type_list + "], out[";
for (size_t input_index = 0; input_index < attr.GetOutputSize(); ++input_index) {
type_list = type_list + TypeId2String(attr.GetOutputAttr(input_index).first) +
((input_index == (attr.GetOutputSize() - 1)) ? "" : " ");
}
type_lists = type_lists + type_list + "]; ";
}
return type_lists;
}
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册