colin
/
mlir-hlo
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

[XLA/GPU] Migrate all unnested elementwise emitters. 

PiperOrigin-RevId: 346658288
This commit is contained in:
A. Unique TensorFlower 2020-12-09 16:04:43 -08:00 committed by TensorFlow MLIR Team
parent 1c10e1fec6
commit 65ebd85563
2 changed files with 0 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
include/mlir-hlo/utils/hlo_utils.h
View File

@ -83,11 +83,6 @@ enum ScalarLimit {
// Requires `ty` to be either FloatType or IntegerType. // Requires `ty` to be either FloatType or IntegerType.
DenseElementsAttr GetScalarLimitOfType(Type ty, ScalarLimit limit); DenseElementsAttr GetScalarLimitOfType(Type ty, ScalarLimit limit);
// Given `op_name` from LMHLO, returns the corresponding op name in MHLO.
// Returns empty string if no such op exists.
std::string LmhloToMhloOpName(llvm::StringRef op_name,
mlir::MLIRContext* context);
} // namespace hlo } // namespace hlo
} // namespace mlir } // namespace mlir

8
lib/utils/hlo_utils.cc
View File

@ -132,13 +132,5 @@ DenseElementsAttr GetScalarLimitOfType(Type ty, ScalarLimit limit) {
llvm_unreachable("unsupported type"); llvm_unreachable("unsupported type");
} }
std::string LmhloToMhloOpName(llvm::StringRef op_name,
mlir::MLIRContext *context) {
assert(op_name.startswith("lmhlo.") && "Expected an LMHLO op");
std::string mhlo_op_name(op_name.drop_front(1));
if (context->isOperationRegistered(mhlo_op_name)) return mhlo_op_name;
return "";
}
} // namespace hlo } // namespace hlo
} // namespace mlir } // namespace mlir