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

register the lax module for stack trace filtering

Browse Source
This commit is contained in:
Roy Frostig 2021-02-22 21:11:41 -08:00
parent 60bff69ee6
commit 4c179d1b66
1 changed files with 5 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
jax/_src/lax/__init__.py
View File

@ -11,3 +11,8 @@
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
import os
from jax._src import traceback_util
traceback_util.register_exclusion(os.path.dirname(__file__))