search

TiledTensor / TiledCUDA

TiledCUDA is a highly efficient kernel template library designed to elevate CUDA C’s level of abstraction for processing tiles.
MIT License
159 stars 10 forks source link

fix(unittest): bug fix for tensor core gemm. #59

Closed haruhi55 closed 5 months ago

haruhi55 commented 5 months ago
  1. Fixed a small bug where slicing the K-dimension used a chunk size smaller than K.
  2. Made a slight improvement to the numeric check for half-precision GEMM by using the average absolute difference instead of a single difference, as the latter is unstable.
  3. Unified the printing style.