Aborted (core dumped) in jt.index_var

[x0w3n]

Describe the bug

If the dtype argument for jt.index_var is invalid, a crash occurs when retrieving the element's index.

Full Log

[i 1202 21:57:22.125618 04 compiler.py:956] Jittor(1.3.9.10) src: /home/miniconda3/envs/jittor/lib/python3.9/site-packages/jittor
[i 1202 21:57:22.135768 04 compiler.py:957] g++ at /usr/bin/g++(11.2.0)
[i 1202 21:57:22.136237 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 21:57:22.202607 04 install_cuda.py:93] cuda_driver_version: [12, 2]
[i 1202 21:57:22.216799 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 21:57:22.492197 04 __init__.py:412] Found gdb(12.0.90) at /usr/bin/gdb.
[i 1202 21:57:22.494810 04 __init__.py:412] Found addr2line(2.38) at /usr/bin/addr2line.
[i 1202 21:57:22.863840 04 compiler.py:1013] cuda key:cu12.2.140_sm_89
[i 1202 21:57:24.081252 04 __init__.py:227] Total mem: 251.50GB, using 16 procs for compiling.
[i 1202 21:57:24.241209 04 jit_compiler.cc:28] Load cc_path: /usr/bin/g++
[i 1202 21:57:24.413111 04 init.cc:63] Found cuda archs: [89,]
Caught segfault at address 0x80, thread_name: '', flush log...
[i 1202 21:57:25.402157 04 tracer.cc:149] stack trace for pid= 341391
[New LWP 341633]
[New LWP 341634]
[New LWP 341635]
[New LWP 341636]
[New LWP 341637]
[New LWP 341638]
[New LWP 341639]
[New LWP 341716]
[New LWP 341717]
[New LWP 341718]
[New LWP 341719]
[New LWP 341720]
[New LWP 341721]
[New LWP 341722]
[New LWP 341723]
[New LWP 341724]
[New LWP 341725]
[New LWP 341726]
[New LWP 341727]
[New LWP 341728]
[New LWP 341729]
[New LWP 341730]
[New LWP 341731]
[Thread debugging using libthread_db enabled]
Using host libthread_db library "/lib/x86_64-linux-gnu/libthread_db.so.1".
0x00007f761c32e49f in __GI___wait4 (pid=341732, stat_loc=0x0, options=0, usage=0x0) at ../sysdeps/unix/sysv/linux/wait4.c:30
30      ../sysdeps/unix/sysv/linux/wait4.c: No such file or directory.
Traceback (most recent call last):
  File "/usr/share/gdb/auto-load/usr/lib/x86_64-linux-gnu/libstdc++.so.6.0.30-gdb.py", line 60, in <module>
    from libstdcxx.v6 import register_libstdcxx_printers
ModuleNotFoundError: No module named 'libstdcxx'
[Current thread is 1 (Thread 0x7f761c23f740 (LWP 341391))]
#0  0x00007f761c32e49f in __GI___wait4 (pid=341732, stat_loc=0x0, options=0, usage=0x0) at ../sysdeps/unix/sysv/linux/wait4.c:30
#1  0x00007f761aadc7f7 in jittor::print_trace() () from /home/.cache/jittor/jt1.3.9/g++11.2.0/py3.9.12/Linux-5.15.0-1x30/INTELRXEONRGOLx51/ef26/default/jit_utils_core.cpython-39-x86_64-linux-gnu.so
#2  0x00007f761aad9a65 in jittor::segfault_sigaction(int, siginfo_t*, void*) () from /home/.cache/jittor/jt1.3.9/g++11.2.0/py3.9.12/Linux-5.15.0-1x30/INTELRXEONRGOLx51/ef26/default/jit_utils_core.cpython-39-x86_64-linux-gnu.so
#3  <signal handler called>
#4  0x00007f76180021b7 in jittor::Var::ptr<int> (this=<optimized out>) at /home/miniconda3/envs/jittor/lib/python3.9/site-packages/jittor/src/var.h:35
#5  jittor::FusedOp::jit_run (this=0x7ffe737b50b0) at /home/.cache/jittor/jt1.3.9/g++11.2.0/py3.9.12/Linux-5.15.0-1x30/INTELRXEONRGOLx51/ef26/default/cu12.2.140_sm_89/jit/__opkey0_index__T_int32__DIM_2__XDIM_2__JIT_1__JIT_cpu_1__graph_010000___var_info__0222__i___hash_7a5adc8dbe7200b7_op.cc:18
#6  0x00007f7618666520 in jittor::Profiler::record_and_run(void (*)(jittor::Op*), jittor::Op*, char const*) () from /home/.cache/jittor/jt1.3.9/g++11.2.0/py3.9.12/Linux-5.15.0-1x30/INTELRXEONRGOLx51/ef26/default/cu12.2.140_sm_89/jittor_core.cpython-39-x86_64-linux-gnu.so
#7  0x00007f761866e12c in jittor::Executor::run_sync(std::vector<jittor::Var*, std::allocator<jittor::Var*> >, bool, bool) () from /home/.cache/jittor/jt1.3.9/g++11.2.0/py3.9.12/Linux-5.15.0-1x30/INTELRXEONRGOLx51/ef26/default/cu12.2.140_sm_89/jittor_core.cpython-39-x86_64-linux-gnu.so
#8  0x00007f7618558414 in jittor::sync(std::vector<jittor::VarHolder*, std::allocator<jittor::VarHolder*> > const&, bool, bool) () from /home/.cache/jittor/jt1.3.9/g++11.2.0/py3.9.12/Linux-5.15.0-1x30/INTELRXEONRGOLx51/ef26/default/cu12.2.140_sm_89/jittor_core.cpython-39-x86_64-linux-gnu.so
#9  0x00007f76185586f3 in jittor::VarHolder::sync(bool, bool) () from /home/.cache/jittor/jt1.3.9/g++11.2.0/py3.9.12/Linux-5.15.0-1x30/INTELRXEONRGOLx51/ef26/default/cu12.2.140_sm_89/jittor_core.cpython-39-x86_64-linux-gnu.so
Undefined command: "py-bt".  Try "help".
[Inferior 1 (process 341391) detached]
Segfault, exit
terminate called without an active exception
Aborted (core dumped)

Minimal Reproduce

import jittor as jt
a = jt.array([[1, 2, 3], [4, 5, 6]], dtype=jt.float32)
k = 2
index_a = jt.index_var(a, k)
print(index_a)

Expected behavior

The dtype parameter needs to be checked to prevent crashes.