Skip to content

Grad error message #721

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.

Sign up for GitHub

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

Closed
wants to merge 17 commits into from
Closed

Grad error message #721

Changes from all commits
Commits
Show all changes
17 commits
Select commit Hold shift + click to select a range
8875fd0
Moved the kron function as-is from slinalg.py to nlinalg.py
tanish1729 Mar 26, 2024
9eb50a6
Updated the kron function to use JAX implementation and moved tests t…
tanish1729 Mar 26, 2024
be52b59
Updated test for kron
tanish1729 Mar 26, 2024
52ed8c3
Updated slinalg to use the new kron implementation
tanish1729 Mar 26, 2024
1611f9a
running all pre commit checks
tanish1729 Mar 26, 2024
5aef097
removed importing reshape
tanish1729 Mar 26, 2024
5299b75
added test for commute with inv
tanish1729 Mar 26, 2024
3e0c79c
updated test
tanish1729 Mar 27, 2024
a7902fd
working test for commute_with_inv
tanish1729 Mar 27, 2024
e3b1aaa
working test for commute_with_inv
tanish1729 Mar 27, 2024
3ea1cdb
skipped some half precision tests
tanish1729 Mar 27, 2024
7fa03cf
skipped some half precision tests
tanish1729 Mar 27, 2024
5104494
updated comment
tanish1729 Mar 27, 2024
35996a5
Parametrized the tests
tanish1729 Mar 27, 2024
869e184
Added grad_not_implemented
tanish1729 Apr 16, 2024
9973b85
Merge branch 'pymc-devs:main' into main
tanish1729 Apr 16, 2024
a354270
Added grad_not_implemented
tanish1729 Apr 17, 2024
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
6 changes: 5 additions & 1 deletion pytensor/graph/op.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -13,6 +13,7 @@

import pytensor
from pytensor.configdefaults import config
from pytensor.gradient import grad_not_implemented
from pytensor.graph.basic import Apply, Variable
from pytensor.graph.utils import (
MetaObject,
Expand Down Expand Up @@ -357,7 +358,10 @@ def grad(
.. [1] Giles, Mike. 2008. “An Extended Collection of Matrix Derivative Results for Forward and Reverse Mode Automatic Differentiation.”

"""
raise NotImplementedError()
# raise NotImplementedError()
Copy link
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

This is my suggestion for a simpler solution

Suggested change
# raise NotImplementedError()
raise NotImplementedError(f"grad not implemented for Op {self}")

Copy link
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Also your implementation was wrong because it assumed there is a single input

(a,) = inputs
a_grad = grad_not_implemented(self, 0, a, comment="To be updated")
return [a_grad]

def L_op(
self,
Expand Down