Skip to content

Commit

Permalink
Fix formatting
Browse files Browse the repository at this point in the history
  • Loading branch information
chsigg authored Aug 23, 2024
1 parent 86a38aa commit f09c4d6
Showing 1 changed file with 4 additions and 4 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -241,13 +241,13 @@ static const std::string S8_to_Bf16_sm90 =
"{ \n"
".reg .b32 l<3>; \n"
".reg .b32 h<3>; \n"
"prmt.b32 l0, $2, 0x43, 0x4140; \n" // Unpack to shifted bf16.
"prmt.b32 l0, $2, 0x43, 0x4140; \n" // Unpack to shifted bf16.
"prmt.b32 h0, $2, 0x43, 0x4342; \n"
"and.b32 l1, l0, 0xff7fff7f; \n" // Zero the least exp bit.
"and.b32 l1, l0, 0xff7fff7f; \n" // Zero the least exp bit.
"and.b32 h1, h0, 0xff7fff7f; \n"
"and.b32 l2, l0, 0xff80ff80; \n" // Zero the mantissa.
"and.b32 l2, l0, 0xff80ff80; \n" // Zero the mantissa.
"and.b32 h2, h0, 0xff80ff80; \n"
"sub.bf16x2 $0, l1, l2; \n" // Subtract the offset.
"sub.bf16x2 $0, l1, l2; \n" // Subtract the offset.
"sub.bf16x2 $1, h1, h2; \n"
"}";

Expand Down

0 comments on commit f09c4d6

Please sign in to comment.