diff --git a/tools/gen_header_for_bin_reduce.py b/tools/gen_header_for_bin_reduce.py index 39fdf3005c2d889fa01e6d73cef5e3b4e5f92d1d..337b83db5436636eeefd3c6ac68a1f709e477362 100755 --- a/tools/gen_header_for_bin_reduce.py +++ b/tools/gen_header_for_bin_reduce.py @@ -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)