blob: 5aad67173cc61fcae18af26f60f69ead786793e3 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
|
//===-------------- TosaTarget.cpp - TOSA Target utilities ----------------===//
//
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
// See https://llvm.org/LICENSE.txt for license information.
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
//
//===----------------------------------------------------------------------===//
#include "mlir/Dialect/Tosa/IR/TargetEnv.h"
namespace mlir {
namespace tosa {
TargetEnvAttr lookupTargetEnv(Operation *op) {
while (op) {
op = SymbolTable::getNearestSymbolTable(op);
if (!op)
break;
if (auto attr = op->getAttrOfType<TargetEnvAttr>(TargetEnvAttr::name))
return attr;
op = op->getParentOp();
}
return {};
}
TargetEnvAttr getDefaultTargetEnv(MLIRContext *context) {
return TargetEnvAttr::get(context, Level::eightK,
{Profile::pro_int, Profile::pro_fp}, {});
}
TargetEnvAttr lookupTargetEnvOrDefault(Operation *op) {
if (auto attr = lookupTargetEnv(op))
return attr;
return getDefaultTargetEnv(op->getContext());
}
} // namespace tosa
} // namespace mlir
|
