• Lei Wang's avatar
    [Refactor] Rename clamp functions and enhance dtype handling in tests (#232) · 7ae35298
    Lei Wang authored
    - Renamed `clamp` to `clamp_within_bounds` and `clamp_v2` to `clamp_value_range` for improved clarity.
    - Updated dtype handling in `clamp_value_range` to use the correct tensor dtype.
    - Modified test cases to reflect the new function names and ensure proper dtype conversion using `map_torch_type`.
    - Enhanced the reference program for clamping to utilize the updated tensor dtype, improving accuracy in tests.
    7ae35298
test_tilelang_language_clamp.py 2.97 KB