mirror of
https://github.com/saymrwulf/onnxruntime.git
synced 2026-06-24 02:47:54 +00:00
ROCM
This commit is contained in:
parent
6017a496f3
commit
ec1efa3dd2
2 changed files with 4 additions and 2 deletions
|
|
@ -15,7 +15,7 @@ namespace cuda {
|
|||
1, \
|
||||
T, \
|
||||
kCudaExecutionProvider, \
|
||||
KernelDefBuilder() \
|
||||
(*KernelDefBuilder::Create()) \
|
||||
.TypeConstraint("V", DataTypeImpl::GetTensorType<T>()) \
|
||||
.TypeConstraint("T", DataTypeImpl::GetTensorType<bool>()), \
|
||||
IsAllFiniteOp<T>);
|
||||
|
|
|
|||
|
|
@ -7,7 +7,9 @@
|
|||
// switching providers to be runnable as shared libraries. The interfaces will become more tightly integrated into the core code.
|
||||
|
||||
#pragma once
|
||||
#ifndef USE_ROCM
|
||||
#ifdef USE_ROCM
|
||||
#include "core/providers/common.h"
|
||||
#else
|
||||
#define SHARED_PROVIDER 1
|
||||
|
||||
#include <vector>
|
||||
|
|
|
|||
Loading…
Reference in a new issue