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

2022-04-14 10:26:48 -07:00
parent 25f6689508
commit dc4d40faec
1 changed files with 1 additions and 0 deletions
--- a/python/triton/code_gen.py
+++ b/python/triton/code_gen.py
@@ -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 = mangled_constants.replace('.', '_d_')
    mangled_constants = mangled_constants.replace("'", '_sq_')
+    mangled_constants = mangled_constants.replace("e-", '_em_')
    ret = f'{name}__{mangled_arg_names}__{mangled_constants}'
    return ret