squall/cutlass
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity
757275f279
cutlass/python/cutlass_library
History
Manish Gupta 757275f279
Adding more Threadblock Tiles for Mixed-input TensorOp (BF16 * S8) in cutlass_library (#1132) 
* Adding more tiles in the cutlass_library for mixed-input support.

* fix rebase issue

* more tiles to upcast a
2023-10-13 11:33:15 -04:00
..
__init__.py CUTLASS 3.2.1 (#1113) 2023-09-26 17:24:26 -04:00
conv2d_operation.py Support for Mixed Input TensorOp (#1084) 2023-09-27 11:18:30 -04:00
conv3d_operation.py Support for Mixed Input TensorOp (#1084) 2023-09-27 11:18:30 -04:00
gemm_operation.py Support for Mixed Input TensorOp (#1084) 2023-09-27 11:18:30 -04:00
generator.py Adding more Threadblock Tiles for Mixed-input TensorOp (BF16 * S8) in cutlass_library (#1132) 2023-10-13 11:33:15 -04:00
library.py Support for Mixed Input TensorOp (#1084) 2023-09-27 11:18:30 -04:00
manifest.py CUTLASS 3.2.1 (#1113) 2023-09-26 17:24:26 -04:00
rank_2k_operation.py Support for Mixed Input TensorOp (#1084) 2023-09-27 11:18:30 -04:00
rank_k_operation.py Support for Mixed Input TensorOp (#1084) 2023-09-27 11:18:30 -04:00
symm_operation.py Support for Mixed Input TensorOp (#1084) 2023-09-27 11:18:30 -04:00
trmm_operation.py Support for Mixed Input TensorOp (#1084) 2023-09-27 11:18:30 -04:00