mirror of
https://github.com/saymrwulf/onnxruntime.git
synced 2026-07-30 20:18:08 +00:00
92dc5c506d
* move all contrib ops to one place * namespace changes * bug fix - remove redundant file after merge master * plus more minor bug fixes * bug fix * fix extra space in include header + namespace fix * fix linux build failure: * fix test group names * remove redundant test
15 lines
No EOL
356 B
C++
15 lines
No EOL
356 B
C++
// Copyright (c) Microsoft Corporation. All rights reserved.
|
|
// Licensed under the MIT License.
|
|
|
|
#pragma once
|
|
#include "core/framework/kernel_registry.h"
|
|
|
|
namespace onnxruntime {
|
|
namespace contrib {
|
|
namespace cuda {
|
|
|
|
void RegisterCudaContribKernels(KernelRegistry& kernel_registry);
|
|
|
|
} // namespace cuda
|
|
} // namespace contrib
|
|
} // namespace onnxruntime
|