如何让SymPy的C99CodePrinter输出带'f'后缀的float字面量？

要让SymPy输出符合OpenCL要求的带'f'后缀的float字面量，你需要子类化C99CodePrinter并重写处理浮点数的方法，具体实现如下：

• 自定义打印机类继承C99CodePrinter，重写_print_Float方法，在浮点数的字符串表示后追加'f'后缀
• 使用自定义打印机实例生成代码

示例代码：

import sympy as sp
from sympy.printing.c import C99CodePrinter
import numpy as np

class OpenCLCodePrinter(C99CodePrinter):
    def _print_Float(self, expr):
        base_str = super()._print_Float(expr)
        # 为带小数点或科学计数法的浮点数添加f后缀
        if '.' in base_str or 'e' in base_str.lower():
            return f"{base_str}f"
        # 处理整数形式的浮点数（如1），转换为1.0f格式
        return f"{base_str}.0f"

# 初始化自定义打印机
my_printer = OpenCLCodePrinter()

x = sp.symbols('x')
y = np.ones(1, dtype=np.float32)
print(my_printer.doprint(y[0] * x))  # 输出：1.0f*x

该方案会统一处理所有浮点数输出：无论是传入的numpy float32对象，还是SymPy自身的Float类型，都会生成带'f'后缀的字面量，完全适配OpenCL的类型规则。

内容的提问来源于stack exchange，提问作者p . clark