GitHub - LeiWang1999/T-MAG: GPU Implementation of T-MAC

GPU Version of T-MAC, implemented with Microsoft BitBLAS and tvm TL.

Requirements

pip install bitblas

Navigation Menu

Repository files navigation

About

Releases

Packages

Contributors

Languages

Name		Name	Last commit message	Last commit date
Latest commit History 2 Commits
group_size2		group_size2
performance		performance
0.lut_gemm.py		0.lut_gemm.py
1.lut_gemm_ntile.py		1.lut_gemm_ntile.py
10.lut_gemm_ntile_ktile_threads_block_reduce_compressed_lop3_kn.py		10.lut_gemm_ntile_ktile_threads_block_reduce_compressed_lop3_kn.py
11.lut_gemm_ntile_ktile_threads_block_reduce_compressed_lop3_reduceisx.py		11.lut_gemm_ntile_ktile_threads_block_reduce_compressed_lop3_reduceisx.py
12.lut_gemm_ntile_ktile_threads_block_reduce_compressed_lop3_reduceisx_shared_cache.py		12.lut_gemm_ntile_ktile_threads_block_reduce_compressed_lop3_reduceisx_shared_cache.py
13.lut_gemm_ntile_ktile_threads_block_reduce_compressed_lop3_lut_shared_local_cache.py		13.lut_gemm_ntile_ktile_threads_block_reduce_compressed_lop3_lut_shared_local_cache.py
2.lut_gemm_ntile_ktile.py		2.lut_gemm_ntile_ktile.py
3.lut_gemm_ntile_ktile_threads.py		3.lut_gemm_ntile_ktile_threads.py
4.lut_gemm_ntile_ktile_threads_block_reduce.py		4.lut_gemm_ntile_ktile_threads_block_reduce.py
5.lut_gemm_ntile_ktile_threads_block_reduce_vectorized.py		5.lut_gemm_ntile_ktile_threads_block_reduce_vectorized.py
6.lut_gemm_ntile_ktile_threads_block_reduce_compressed.py		6.lut_gemm_ntile_ktile_threads_block_reduce_compressed.py
7.lut_gemm_ntile_ktile_threads_block_reduce_compressed_lop3.py		7.lut_gemm_ntile_ktile_threads_block_reduce_compressed_lop3.py
8.lut_gemm_ntile_ktile_threads_block_reduce_compressed_lop3_lut_local_cache.py		8.lut_gemm_ntile_ktile_threads_block_reduce_compressed_lop3_lut_local_cache.py
9.lut_gemm_ntile_ktile_threads_block_reduce_compressed_lop3_lut_shared_cache.py		9.lut_gemm_ntile_ktile_threads_block_reduce_compressed_lop3_lut_shared_cache.py
README.md		README.md

Folders and files

Latest commit

History

Repository files navigation

About

Resources

Stars

Watchers

Forks

Releases

Packages

Contributors

Languages