Skip to content

Conversation

@LeiWang1999
Copy link
Member

For roadmap #6 , Tthe most important changes include adding a new function for tensor conversion, updating licensing information in multiple files, and improving error messages and tensor handling.

Tensor conversion and testing enhancements:

  • testing/python/kernel/test_tilelang_dequantize_gemm.py: Added new functions _tir_u8_to_f4_to_f16, torch_convert, _convert_test, test_fp4_fp16_convert_close, matmul_fp16xfp4, ref_program, assert_simple_impl_float16xfp4_gemm, and test_simple_impl_float16xfp4_gemm to support and test tensor conversion from uint8 to float16 and matrix multiplication.

Code improvements:

  • tilelang/jit/kernel.py: Enhanced the error message for invalid execution backend by including the provided backend in the message.
  • tilelang/utils/tensor.py: Improved tensor generation by adding a check for unsigned integers and adjusting the range for integer tensor values accordingly.

Miscellaneous:

  • .gitattributes: Added a new entry to classify .h files as C++ language for linguist.

@LeiWang1999 LeiWang1999 merged commit f4df38d into tile-ai:main Jan 23, 2025
4 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant