Skip to content

Commit

Permalink
Duplicate dequant nodes for dq encoder (pytorch#4222)
Browse files Browse the repository at this point in the history
Summary:
Pull Request resolved: pytorch#4222

{F1749810435}

Here, slice_9 and slice_10 are nops. When eliminated, the graph reduces to

{F1749810958}

dequantize_per_tensor_tensor_3 becomes input to two linear ops, causing failure in dq partitioner. This diff creates duplicates of dequant nodes so that each linear has unique inputs for dq partitioner.

Reviewed By: larryliu0820, pssrawat

Differential Revision: D59647685

fbshipit-source-id: 51ac80b11ce855a5eee2746c815d2f882beaa520
  • Loading branch information
tarun292 authored and facebook-github-bot committed Jul 12, 2024
1 parent b7df20d commit e453652
Showing 1 changed file with 1 addition and 0 deletions.
1 change: 1 addition & 0 deletions backends/transforms/TARGETS
Original file line number Diff line number Diff line change
Expand Up @@ -120,6 +120,7 @@ runtime.python_library(
"//executorch/backends/...",
"//executorch/examples/...",
"//executorch/extension/llm/...",
"@EXECUTORCH_CLIENTS",
],
deps = [
"//caffe2:torch",
Expand Down

0 comments on commit e453652

Please sign in to comment.