diff --git a/python/triton/code_gen.py b/python/triton/code_gen.py index 7553bee55..5fd1c1be6 100644 --- 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