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

ROCM

Browse source
This commit is contained in:
Ryan Hill 2021-04-15 15:18:39 -07:00
parent 6662d8d66d
commit 9c6cc88e82
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/core/providers/shared_library/provider_api.h
View file

@ -10,6 +10,7 @@
#ifdef USE_ROCM
#include "core/providers/common.h"
#include "core/providers/cpu/tensor/onehot.h"
#include "core/providers/cpu/tensor/gather_elements.h"
#else
#define SHARED_PROVIDER 1