Skip to content
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.

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

Segmentation fault (core dumped) in jt.nn.relu #610

Open
x0w3n opened this issue Dec 2, 2024 · 0 comments
Open

Segmentation fault (core dumped) in jt.nn.relu #610

x0w3n opened this issue Dec 2, 2024 · 0 comments

Comments

@x0w3n
Copy link

x0w3n commented Dec 2, 2024

Describe the bug

A crash is triggered when the input value of jt.nn.relu is empty.

Full Log

[i 1202 22:33:21.922346 04 compiler.py:956] Jittor(1.3.9.10) src: /home/miniconda3/envs/jittor/lib/python3.9/site-packages/jittor
[i 1202 22:33:21.932980 04 compiler.py:957] g++ at /usr/bin/g++(11.2.0)
[i 1202 22:33:21.933480 04 compiler.py:958] cache_path: /home/.cache/jittor/jt1.3.9/g++11.2.0/py3.9.12/Linux-5.15.0-1x30/INTELRXEONRGOLx51/ef26/default
[i 1202 22:33:22.002532 04 install_cuda.py:93] cuda_driver_version: [12, 2]
[i 1202 22:33:22.016911 04 __init__.py:412] Found /home/.cache/jittor/jtcuda/cuda12.2_cudnn8_linux/bin/nvcc(12.2.140) at /home/.cache/jittor/jtcuda/cuda12.2_cudnn8_linux/bin/nvcc.
[i 1202 22:33:22.461689 04 __init__.py:412] Found gdb(12.0.90) at /usr/bin/gdb.
[i 1202 22:33:22.464139 04 __init__.py:412] Found addr2line(2.38) at /usr/bin/addr2line.
[i 1202 22:33:22.856455 04 compiler.py:1013] cuda key:cu12.2.140_sm_89
[i 1202 22:33:23.853469 04 __init__.py:227] Total mem: 251.50GB, using 16 procs for compiling.
[i 1202 22:33:24.027103 04 jit_compiler.cc:28] Load cc_path: /usr/bin/g++
[i 1202 22:33:24.256860 04 init.cc:63] Found cuda archs: [89,]
Segmentation fault (core dumped)

Minimal Reproduce

import jittor as jt
from jittor import nn
import sys
input_tensor = jt.array([])

m = nn.relu(input_tensor)

Expected behavior

The input values of the activation function need to be checked to prevent crashes.

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

No branches or pull requests

1 participant