tilelang

trending/tilelang

Fork 0

mirror of https://github.com/tile-ai/tilelang.git synced 2026-10-22 06:50:25 +08:00

Files

T

History

Tong WUandZhiwen Mo 11456de25f [Feature] Add tl.infinity operator for infinity handling of bfloat16 (#1175 )

* Update dependency version for apache-tvm-ffi in pyproject.toml to fix CI

* [Math] Add `tl.infinity` operation and update Python interface for infinity handling

- Implemented `infinity_op` in C++ to return infinity values for supported data types.
- Registered new operation `tl.infinity` with appropriate attributes.
- Updated Python interface to call the new `tl.infinity` operation instead of the previous method.

* Add unit tests for `tl.infinity` operation in TileLang

- Introduced a new test file `test_tilelang_language_infinity.py` to validate the behavior of the `tl.infinity` operation across multiple data types (float16, bfloat16, float32, float64).
- Implemented a kernel to fill a tensor with infinity values and asserted the correctness of the output against PyTorch's `torch.inf`.

* lint

---------

Co-authored-by: Zhiwen Mo <zm125@ic.ac.uk>

2025-11-06 01:22:29 +08:00

amd

[Refactor] Improve Python3.9 compatibility for ParamSpec and Self (#1190 )

2025-11-04 17:05:58 +08:00

autotune

[Autotune] Add autotune coverage for symbolic M and normalize cache key (#1075 )

2025-10-20 17:40:04 +08:00

cache

[Refactor] Refactor barrier management (#744 )

2025-08-21 20:03:05 +08:00

carver

[Refactor] Phaseout LLVM Dependency by Making it Optional (#247 )