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

Check failed in jt.scatter/jt.random #618

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

Check failed in jt.scatter/jt.random #618

x0w3n opened this issue Dec 3, 2024 · 0 comments

Comments

@x0w3n
Copy link

x0w3n commented Dec 3, 2024

Describe the bug

Check failed is triggered if the jt.scatter reduce parameter is not supported.
The same error occurs at jt.random.
The error log suggests reporting this issue.

Full Log

[i 1203 11:01:28.290158 44 compiler.py:956] Jittor(1.3.9.10) src: /home/miniconda3/envs/jittor/lib/python3.9/site-packages/jittor
[i 1203 11:01:28.297562 44 compiler.py:957] g++ at /usr/bin/g++(11.2.0)
[i 1203 11:01:28.297614 44 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 1203 11:01:28.380672 44 install_cuda.py:93] cuda_driver_version: [12, 2]
[i 1203 11:01:28.394788 44 __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 1203 11:01:28.587427 44 __init__.py:412] Found gdb(12.0.90) at /usr/bin/gdb.
[i 1203 11:01:28.589898 44 __init__.py:412] Found addr2line(2.38) at /usr/bin/addr2line.
[i 1203 11:01:28.959561 44 compiler.py:1013] cuda key:cu12.2.140_sm_89
[i 1203 11:01:30.272529 44 __init__.py:227] Total mem: 251.50GB, using 16 procs for compiling.
[i 1203 11:01:30.463344 44 jit_compiler.cc:28] Load cc_path: /usr/bin/g++
[i 1203 11:01:30.735148 44 init.cc:63] Found cuda archs: [89,]
Traceback (most recent call last):
  File "test.py", line 7, in <module>
    jt.scatter(x, 0, index, src,reduce = 'test')
  File "/home/miniconda3/envs/jittor/lib/python3.9/site-packages/jittor/misc.py", line 1608, in scatter
    return x.setitem(tuple(indexes), src, reduce)
RuntimeError: Wrong inputs arguments, Please refer to examples(help(jt.setitem)).

Types of your inputs are:
 self   = Var,
 args   = (tuple, Var, str, ),

The function declarations are:
 VarHolder* setitem(VarHolder* x,  VarSlices&& slices, VarHolder* y,  NanoString op=ns_void)

Failed reason:[f 1203 11:01:31.545916 44 nano_string.h:163] Check failed: iter != __string_to_ns.end()  Something wrong... Could you please report this issue?
 hack

Minimal Reproduce

import jittor as jt

src = jt.arange(1, 11).reshape((2, 5))
index = jt.array([[0, 1, 2, 0]])
x = jt.zeros((3, 5), dtype=src.dtype)
jt.scatter(x, 0, index, src,reduce = 'hack')
import jittor as jt
c = jt.random(shape=(3,), type='hack')

Expected behavior

The reduce/type parameters of jt.scatter need to be checked.

@x0w3n x0w3n changed the title Check failed in jt.scatter Check failed in jt.scatter/jt.random Dec 3, 2024
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