NVIDIA/TensorRT-Incubator

stablehlo.select : failed to determine expected shape

Closed this issue · 1 comments

tester.mlir:9231:13: error: failed to determine expected shape
    %8704 = stablehlo.select %8701, %8702, %8703 : tensor<?x?xi1>, tensor<?x?xf16>
            ^
tester.mlir:9231:13: error: 'tensorrt.select' op failed to infer returned types
    %8704 = stablehlo.select %8701, %8702, %8703 : tensor<?x?xi1>, tensor<?x?xf16>
            ^
tester.mlir:9231:13: note: see current operation: %876 = "tensorrt.select"(%870, %875, %14) : (tensor<?x?xi1>, tensor<?x?xf16>, tensor<1x1xf16>) -> tensor<?x?xf16>

selectfailure.txt

Fixed by #380 and #376