Skip to content
Commit c020480f authored by Lei Zhang's avatar Lei Zhang Committed by A. Unique TensorFlower
Browse files

[spirv] Allow return ops to be in control flow ops 

Use `getParentOfType<FunctionOp>()` instead of `cast<FuncOp>(getParentOp())`
to avoid crash when return ops are used inside spv.selection/spv.loop.

PiperOrigin-RevId: 273006041
parent 58e2ead3
Show whitespace changes
Inline Side-by-side
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Please to comment