rocm_ops.h 280 Bytes
Newer Older
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
// Copyright (c) Microsoft Corporation. All rights reserved.
// Licensed under the MIT License.

#pragma once

namespace Rocm {

#ifdef USE_ROCM

void RegisterOps(Ort::CustomOpDomain& domain);

#else

inline void RegisterOps(Ort::CustomOpDomain&) {}

#endif

}  // namespace Rocm