From dc4d40faec8fdb8b9d46ab2e36c303a8d3e34f29 Mon Sep 17 00:00:00 2001 From: Philippe Tillet Date: Thu, 14 Apr 2022 10:26:48 -0700 Subject: [PATCH] [FRONTEND] now mangle constexpr float containing "e-" --- python/triton/code_gen.py | 1 + 1 file changed, 1 insertion(+) 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