mirror of
https://github.com/saymrwulf/onnxruntime.git
synced 2026-07-29 20:14:01 +00:00
7b39f5090c
* Add Attention op for multi head self attention in BERT * Add test cases * Move op from kOnnxDomain to kMSDomain. Limit test to run by CUDA provider only. * fix test * Add float16 test * fix cpu build error * handle cuda error * get last cuda error when failed
21 lines
1 KiB
C++
21 lines
1 KiB
C++
// Copyright (c) Microsoft Corporation. All rights reserved.
|
|
// Licensed under the MIT License.
|
|
#pragma once
|
|
#include "core/framework/execution_provider.h"
|
|
|
|
namespace onnxruntime {
|
|
namespace test {
|
|
|
|
// unique_ptr providers with default values for session registration
|
|
std::unique_ptr<IExecutionProvider> DefaultCpuExecutionProvider(bool enable_arena = true);
|
|
std::unique_ptr<IExecutionProvider> DefaultCudaExecutionProvider();
|
|
std::unique_ptr<IExecutionProvider> DefaultMkldnnExecutionProvider(bool enable_arena = true);
|
|
std::unique_ptr<IExecutionProvider> DefaultNGraphExecutionProvider();
|
|
std::unique_ptr<IExecutionProvider> DefaultNupharExecutionProvider(bool allow_unaligned_buffers = true);
|
|
std::unique_ptr<IExecutionProvider> DefaultBrainSliceExecutionProvider();
|
|
std::unique_ptr<IExecutionProvider> DefaultTensorrtExecutionProvider();
|
|
std::unique_ptr<IExecutionProvider> DefaultOpenVINOExecutionProvider();
|
|
std::unique_ptr<IExecutionProvider> DefaultNnapiExecutionProvider();
|
|
|
|
} // namespace test
|
|
} // namespace onnxruntime
|