mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-19 05:16:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

Internal change

Browse Source 
PiperOrigin-RevId: 607803148
This commit is contained in:
Rebecca Chen 2024-02-16 22:32:41 +00:00 committed by jax authors
parent 772743e6a4
commit 6b7dd6ff38
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
jax/_src/interpreters/mlir.py
View File

@ -696,7 +696,7 @@ class LoweringRuleContext:
assert self.tokens_out is None, 'Should only set `tokens_out` once.'
self.tokens_out = tokens_out
def replace(self, **kw): return dataclasses.replace(self, **kw)
def replace(self, **kw): return dataclasses.replace(self, **kw) # pytype: disable=wrong-arg-types # dataclasses-replace-types
if not MYPY: