Skip to content

Commit 246ea29

Browse files
committed
Add comment
1 parent feaed53 commit 246ea29

File tree

1 file changed

+5
-0
lines changed

1 file changed

+5
-0
lines changed

src/relay/qnn/op/rsqrt.cc

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -78,6 +78,11 @@ Expr MakeQuantizedRsqrt(Expr x, Expr scale, Expr zero_point, Expr output_scale,
7878
*/
7979
Expr QnnRsqrtCanonicalize(const Attrs& attrs, const Array<Expr>& new_args,
8080
const Array<tvm::relay::Type>& arg_types) {
81+
// At this time, due to the complexity of implementing this op in int8 or uint8,
82+
// we dequantize the input, run the op in float, and then quantize the output (as below).
83+
// This acts as a placeholder for future hardware enablement, where more hardware specific
84+
// canonicalization can be provided.
85+
8186
// Get the args.
8287
QnnUnaryOpArguments args(new_args);
8388

0 commit comments

Comments
 (0)