gaspersic/triton
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

[FRONTEND] now mangle constexpr float containing "e-"

Browse Source
This commit is contained in:
Philippe Tillet
2022-04-14 10:26:48 -07:00
parent 25f6689508
commit dc4d40faec
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
python/triton/code_gen.py
View File

@@ -61,6 +61,7 @@ def mangle_fn(name, arg_tys, constants):
mangled_constants = '_'.join([f'{i}c{key(constants[i])}' for i in sorted(constants)]) mangled_constants = '_'.join([f'{i}c{key(constants[i])}' for i in sorted(constants)])
mangled_constants = mangled_constants.replace('.', '_d_') mangled_constants = mangled_constants.replace('.', '_d_')
mangled_constants = mangled_constants.replace("'", '_sq_') mangled_constants = mangled_constants.replace("'", '_sq_')
mangled_constants = mangled_constants.replace("e-", '_em_')
ret = f'{name}__{mangled_arg_names}__{mangled_constants}' ret = f'{name}__{mangled_arg_names}__{mangled_constants}'
return ret return ret