Skip to content

Commit 9a3ece2

Browse files
committed
[mlir][sparse] Fix the calling convention of __truncsfbf2 on windows x64
It also wants us to return the value in XMM0.
1 parent 9f433bf commit 9a3ece2

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

mlir/lib/ExecutionEngine/Float16bits.cpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -165,7 +165,7 @@ bool operator==(const bf16 &f1, const bf16 &f2) { return f1.bits == f2.bits; }
165165
#endif
166166
#endif
167167

168-
#if defined(__x86_64__)
168+
#if defined(__x86_64__) || defined(_M_X64)
169169
// On x86 bfloat16 is passed in SSE registers. Since both float and __bf16
170170
// are passed in the same register we can use the wider type and careful casting
171171
// to conform to x86_64 psABI. This only works with the assumption that we're

0 commit comments

Comments
 (0)