Skip to content

[AutoDiff] Destroy unused pullback direct results. #28207

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
merged 1 commit into from
Nov 12, 2019

Conversation

dan-zheng
Copy link
Contributor

@dan-zheng dan-zheng commented Nov 12, 2019

In PullbackEmitter::visitApplyInst, destroy unused pullback direct results.
This is needed for VJPs extracted from @differentiable function callees,
where the @differentiable function's differentiation parameter indices are
a superset of the active apply parameter indices.

Resolves TF-928.

@dan-zheng dan-zheng added the tensorflow This is for "tensorflow" branch PRs. label Nov 12, 2019
@dan-zheng dan-zheng requested review from rxwei and bgogul November 12, 2019 05:02
@dan-zheng
Copy link
Contributor Author

@swift-ci Please test tensorflow

@dan-zheng dan-zheng changed the title [AutoDiff] Consume unused pullback direct results. [AutoDiff] Destroy unused pullback direct results. Nov 12, 2019
In `PullbackEmitter::visitApplyInst`, destroy unused pullback direct results.
This is needed for VJPs extracted from `@differentiable` function callees,
where the `@differentiable` function's differentiation parameter indices are
a superset of the active `apply` parameter indices.

Resolves TF-953.
@dan-zheng
Copy link
Contributor Author

@swift-ci Please test tensorflow

@dan-zheng dan-zheng merged commit 317ef49 into swiftlang:tensorflow Nov 12, 2019
@dan-zheng dan-zheng deleted the TF-953 branch November 12, 2019 15:16
bgogul pushed a commit that referenced this pull request Nov 19, 2019
In `PullbackEmitter::visitApplyInst`, destroy unused pullback direct results.
This is needed for VJPs extracted from `@differentiable` function callees,
where the `@differentiable` function's differentiation parameter indices are
a superset of the active `apply` parameter indices.

Resolves TF-953.
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
tensorflow This is for "tensorflow" branch PRs.
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants