提交 f3ed59d3 编写于 作者: M Megvii Engine Team

feat(dnn/opencl): add heuristic rule for elemwise

GitOrigin-RevId: 2bc574b5a73883356cbb468d2f025a9f6ced0d06
上级 29d24dbb
......@@ -185,7 +185,7 @@ class HeaderGen:
).stdout.decode('utf-8')
self._fout.write('// midout \n')
self._fout.write(cvt)
if cvt.find(" half,"):
if cvt.find(" half,") > 0:
change = open(self._fout.name).read().replace(" half,", " __fp16,")
with open("fix_fp16_bin_reduce.h", "w") as fix_fp16:
fix_fp16.write(change)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册