This commit is contained in:
Ryan Hill 2021-04-15 14:41:58 -07:00
parent 6017a496f3
commit ec1efa3dd2
2 changed files with 4 additions and 2 deletions

View file

@ -15,7 +15,7 @@ namespace cuda {
1, \
T, \
kCudaExecutionProvider, \
KernelDefBuilder() \
(*KernelDefBuilder::Create()) \
.TypeConstraint("V", DataTypeImpl::GetTensorType<T>()) \
.TypeConstraint("T", DataTypeImpl::GetTensorType<bool>()), \
IsAllFiniteOp<T>);

View file

@ -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>