• Lei Wang's avatar
    [Enhancement] Extend pythonic_expr to support dtype mapping in utils.py (#641) · 60974197
    Lei Wang authored
    - Updated the `pythonic_expr` function to accept an optional `dtype_map` parameter, allowing for more flexible type conversions.
    - Refactored calls to `pythonic_expr` in `TLCUDASourceWrapper` to utilize the new mapping feature, improving type handling in kernel generation.
    - Enhanced code clarity by consolidating repeated calls to `pythonic_expr` into a private method within the wrapper class.
    60974197
wrapper.py 43.4 KB