Skip to content

Commit 0849424

Browse files
[JAX] Re-enable grad(sin(x)) test case in xla_metadata_test.
PiperOrigin-RevId: 831472679
1 parent e23706f commit 0849424

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

tests/xla_metadata_test.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -390,8 +390,7 @@ def wrapped_fn(x):
390390

391391
@parameterized.parameters(
392392
("x*x", lambda x: x * x, "add"),
393-
# TODO(b/459818130): Re-enable once stablehlo changes (cl/797055546) are on HEAD.
394-
# ("sin(x)", jnp.sin, "cosine"),
393+
("sin(x)", jnp.sin, "cosine"),
395394
("tanh(x)", jnp.tanh, "add"),
396395
("1/x", lambda x: 1 / x, "negate"),
397396
("sinc(x)", jnp.sinc, "call"),

0 commit comments

Comments
 (0)