saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-05-23 22:13:38 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

exclude packed_attention* from rocm (#15161)

Browse source 
exclude Contrib op PackedAttention from ROCM EP
This commit is contained in:
Yufeng Li 2023-03-22 22:58:57 -07:00 committed by GitHub
parent fc3a2a3771
commit dccbe9d492
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
cmake/onnxruntime_rocm_hipify.cmake
View file

@ -27,6 +27,10 @@ set(contrib_ops_excluded_files
"bert/tensorrt_fused_multihead_attention/*"
"bert/transformer_common.h"
"bert/transformer_common.cc"
"bert/packed_attention.h"
"bert/packed_attention.cc"
"bert/packed_attention_impl.h"
"bert/packed_attention_impl.cu"
"diffusion/group_norm_impl.cu"
"diffusion/nhwc_conv.cc"
"math/complex_mul.cc"