mirror of
https://github.com/tensorflow/tensorflow.git
synced 2026-09-28 13:23:37 +08:00
Adrian Kuegel
c85ab3fa08
Upstream LLVM removed PointerUnion::get<T>() and is<T>(). This patches tf_runtime to use LLVM cast<T>() in basic_kernels.cc for OSS builds. PiperOrigin-RevId: 971845980
18 lines
757 B
Diff
18 lines
757 B
Diff
diff --git a/lib/basic_kernels/opdefs/basic_kernels.cc b/lib/basic_kernels/opdefs/basic_kernels.cc
|
|
--- a/lib/basic_kernels/opdefs/basic_kernels.cc
|
|
+++ b/lib/basic_kernels/opdefs/basic_kernels.cc
|
|
@@ -137,11 +137,11 @@
|
|
void CallOp::setCalleeFromCallable(CallInterfaceCallable callee) {
|
|
// Direct call.
|
|
if (FlatSymbolRefAttr calleeAttr = getCalleeAttr()) {
|
|
- auto symRef = callee.get<SymbolRefAttr>();
|
|
+ auto symRef = cast<SymbolRefAttr>(callee);
|
|
return setCalleeAttr(cast<FlatSymbolRefAttr>(symRef));
|
|
}
|
|
// Indirect call, callee Value is the first operand.
|
|
- return setOperand(0, callee.get<Value>());
|
|
+ return setOperand(0, cast<Value>(callee));
|
|
}
|
|
|
|
//===----------------------------------------------------------------------===//
|