Skip to content

Suppress subsample log_prob JIT warning#3455

Draft
Mirochill wants to merge 1 commit into
pyro-ppl:devfrom
Mirochill:fix-3449-subsample-jit-warning
Draft

Suppress subsample log_prob JIT warning#3455
Mirochill wants to merge 1 commit into
pyro-ppl:devfrom
Mirochill:fix-3449-subsample-jit-warning

Conversation

@Mirochill

Copy link
Copy Markdown

Proposed changes

  • Reuse ignore_jit_warnings around the constant zero tensor created by _Subsample.log_prob.
  • Share the warning filter so it covers both orch.tensor and orch.Tensor message variants.
  • Add JIT regression coverage for _Subsample.log_prob.

Related issue

Fixes #3449

Tests

Not run locally.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

[bug] TracerWarning from subsample_messenger when running NUTS with jit_compile=True

1 participant