Skip to content

Commit

Permalink
Fix incorrect bias stride in matmul cutlass offload (apache#212)
Browse files Browse the repository at this point in the history
This PR makes the cutlass codegen use the correct bias stride when bias
has more than 2 dimensions. For example, if the input bias has shape (1,
n, 4096), the original code will set `ldc` to 0, which produces
incorrect result.

cc @vinx13 @masahi
  • Loading branch information
yelite authored and tqchen committed May 20, 2023
1 parent 29f6e81 commit e787ef4
Showing 0 changed files with 0 additions and 0 deletions.

0 comments on commit e787ef4

Please sign in to comment.