saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-07-25 19:48:11 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Fix py package pipeline (#20065)

Browse source 
### Description
<!-- Describe your changes. -->



### Motivation and Context
Fixes #20068
This commit is contained in:
Ye Wang 2024-03-27 15:59:35 -07:00 committed by GitHub
parent b95fd4e644
commit a9d9b083e4
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 6 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
onnxruntime/contrib_ops/cuda/moe/ft_moe/moe_kernel.cu
View file

@ -656,7 +656,9 @@ inline __device__ float4 operator*(const float4 a, const float4 b) {
return make_float4(a.x * b.x, a.y * b.y, a.z * b.z, a.w * b.w);
}
#if defined(__CUDA_ARCH__) && __CUDA_ARCH__ < 530
// TODO(wy): use cuda common header and investigate pipeline build issue.
#if defined(__CUDA_ARCH__) && __CUDA_ARCH__ < 530 && \
((__CUDACC_VER_MAJOR__ < 12) || ((__CUDACC_VER_MAJOR__ == 12) && (__CUDACC_VER_MINOR__ < 2)))
inline __device__ half operator*(const half a, const half b) {
return __float2half(__half2float(a) * __half2float(b));
}
@ -666,8 +668,10 @@ inline __device__ half2 operator*(const half2 a, const half2 b) {
}
#endif
// TODO(wy): use cuda common header and investigate pipeline build issue.
inline __device__ Half4 operator*(const Half4 a, const Half4 b) {
#if defined(__CUDA_ARCH__) && __CUDA_ARCH__ < 530
#if defined(__CUDA_ARCH__) && __CUDA_ARCH__ < 530 && \
((__CUDACC_VER_MAJOR__ < 12) || ((__CUDACC_VER_MAJOR__ == 12) && (__CUDACC_VER_MINOR__ < 2)))
Half4 result;
result.x = a.x * b.x;
result.y = a.y * b.y;