Check failed in jt.distributions.GammaDistribution

[x0w3n]

Describe the bug

A Check failed was triggered when the sample method of jt.distributions.GammaDistribution was called. The error log suggests reporting this issue.

Full Log

[i 1203 10:07:39.514123 00 compiler.py:956] Jittor(1.3.9.10) src: /home/miniconda3/envs/jittor/lib/python3.9/site-packages/jittor
[i 1203 10:07:39.524758 00 compiler.py:957] g++ at /usr/bin/g++(11.2.0)
[i 1203 10:07:39.525211 00 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 10:07:39.587608 00 install_cuda.py:93] cuda_driver_version: [12, 2]
[i 1203 10:07:39.598939 00 __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 10:07:39.778228 00 __init__.py:412] Found gdb(12.0.90) at /usr/bin/gdb.
[i 1203 10:07:39.780821 00 __init__.py:412] Found addr2line(2.38) at /usr/bin/addr2line.
[i 1203 10:07:40.152362 00 compiler.py:1013] cuda key:cu12.2.140_sm_89
[i 1203 10:07:41.310891 00 __init__.py:227] Total mem: 251.50GB, using 16 procs for compiling.
[i 1203 10:07:41.503042 00 jit_compiler.cc:28] Load cc_path: /usr/bin/g++
[i 1203 10:07:41.771545 00 init.cc:63] Found cuda archs: [89,]
Traceback (most recent call last):
  File "test.py", line 11, in <module>
    print(samples)
  File "/home/miniconda3/envs/jittor/lib/python3.9/site-packages/jittor/__init__.py", line 2088, in vtos
    data_str = f"jt.Var({v.numpy()}, dtype={v.dtype})"
RuntimeError: Wrong inputs arguments, Please refer to examples(help(jt.numpy)).

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

The function declarations are:
 ArrayArgs fetch_sync()

Failed reason:[f 1203 10:07:42.574251 00 op.cc:183] Check failed: flags.get(NodeFlags::_cpu)  Something wrong... Could you please report this issue?
 Op code doesn't have cpu version

Minimal Reproduce

import jittor as jt

concentration = 2.5
rate = 1.2
gamma_dist = jt.distributions.GammaDistribution(concentration, rate)
samples = gamma_dist.sample((10,))

print(samples)

Expected behavior