llvm
/
torch-mlir
mirror of https://github.com/llvm/torch-mlir
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Minor change in TMTensorOps.td (#3602) 

Fixed a little programming choice style that bothered me.
pull/3604/head
rohan-tan-bhowmik 2024-08-14 04:03:49 -07:00 committed by GitHub
parent 4a0bed0ce0
commit 1c16de147a
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
include/torch-mlir-dialects/Dialect/TMTensor/IR/TMTensorOps.td
View File

@ -373,9 +373,8 @@ def TMTensor_TopkOp : TMTensor_Op<"topk",
std::optional<Value> indices() {
if (getNumInputs() < 2) {
return {};
} else {
return getInputOperand(1)->get();
}
return getInputOperand(1)->get();
}
Value outputValues() {
return getOutputOperand(0)->get();