saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-07-30 20:18:08 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Add extra include to fix build w/ CUDA 12 (#14659)

Browse source 
Signed-off-by: Cliff Woolley <jwoolley@nvidia.com>

### Description
Including file to fix build w/CUDA 12



### Motivation and Context
It should allow users to compile against CUDA 12

Signed-off-by: Cliff Woolley <jwoolley@nvidia.com>
Co-authored-by: Cliff Woolley <jwoolley@nvidia.com>
This commit is contained in:
Misha Chornyi 2023-02-11 16:14:40 -08:00 committed by GitHub
parent ba00f3a134
commit 12d91173c4
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

1
onnxruntime/contrib_ops/cuda/bert/attention_impl.cu
View file

@ -28,6 +28,7 @@ limitations under the License.
#include <cassert>
#include <cuda_fp16.h>
#include <cub/cub.cuh>
#include "core/providers/cuda/cu_inc/common.cuh"
#include "core/providers/cuda/cuda_common.h"
#include "core/providers/cuda/shared_inc/fpgeneric.h"