LICENSE.txt
MANIFEST.in
README.md
setup.py
python/jittor/__init__.py
python/jittor/attention.py
python/jittor/compile_extern.py
python/jittor/compiler.py
python/jittor/contrib.py
python/jittor/depthwise_conv.py
python/jittor/distributions.py
python/jittor/init.py
python/jittor/init_cupy.py
python/jittor/linalg.py
python/jittor/lr_scheduler.py
python/jittor/misc.py
python/jittor/nn.py
python/jittor/optim.py
python/jittor/pool.py
python/jittor/pyjt_compiler.py
python/jittor/sparse.py
python/jittor/version
python/jittor.egg-info/PKG-INFO
python/jittor.egg-info/SOURCES.txt
python/jittor.egg-info/dependency_links.txt
python/jittor.egg-info/requires.txt
python/jittor.egg-info/top_level.txt
python/jittor/dataset/__init__.py
python/jittor/dataset/cifar.py
python/jittor/dataset/dataset.py
python/jittor/dataset/mnist.py
python/jittor/dataset/sampler.py
python/jittor/dataset/utils.py
python/jittor/dataset/voc.py
python/jittor/extern/cuda/cub/inc/cub_test.h
python/jittor/extern/cuda/cub/ops/cub_arg_reduce_op.cc
python/jittor/extern/cuda/cub/ops/cub_arg_reduce_op.h
python/jittor/extern/cuda/cub/ops/cub_argsort_op.cc
python/jittor/extern/cuda/cub/ops/cub_argsort_op.h
python/jittor/extern/cuda/cub/ops/cub_cumsum_op.cc
python/jittor/extern/cuda/cub/ops/cub_cumsum_op.h
python/jittor/extern/cuda/cub/ops/cub_test_op.cc
python/jittor/extern/cuda/cub/ops/cub_test_op.h
python/jittor/extern/cuda/cub/ops/cub_where_op.cc
python/jittor/extern/cuda/cub/ops/cub_where_op.h
python/jittor/extern/cuda/cublas/inc/cublas_warper.h
python/jittor/extern/cuda/cublas/ops/cublas_batched_matmul_op.cc
python/jittor/extern/cuda/cublas/ops/cublas_batched_matmul_op.h
python/jittor/extern/cuda/cublas/ops/cublas_matmul_op.cc
python/jittor/extern/cuda/cublas/ops/cublas_matmul_op.h
python/jittor/extern/cuda/cublas/ops/cublas_test_op.cc
python/jittor/extern/cuda/cublas/ops/cublas_test_op.h
python/jittor/extern/cuda/cublas/src/cublas_matmul_test.cc
python/jittor/extern/cuda/cublas/src/cublas_warper.cc
python/jittor/extern/cuda/cublas/src/helper_cublas.cc
python/jittor/extern/cuda/cudnn/inc/cudnn_rnn_descriptor.h
python/jittor/extern/cuda/cudnn/inc/cudnn_warper.h
python/jittor/extern/cuda/cudnn/ops/cudnn_conv3d_backward_w_op.cc
python/jittor/extern/cuda/cudnn/ops/cudnn_conv3d_backward_w_op.h
python/jittor/extern/cuda/cudnn/ops/cudnn_conv3d_backward_x_op.cc
python/jittor/extern/cuda/cudnn/ops/cudnn_conv3d_backward_x_op.h
python/jittor/extern/cuda/cudnn/ops/cudnn_conv3d_op.cc
python/jittor/extern/cuda/cudnn/ops/cudnn_conv3d_op.h
python/jittor/extern/cuda/cudnn/ops/cudnn_conv_backward_w_op.cc
python/jittor/extern/cuda/cudnn/ops/cudnn_conv_backward_w_op.h
python/jittor/extern/cuda/cudnn/ops/cudnn_conv_backward_x_op.cc
python/jittor/extern/cuda/cudnn/ops/cudnn_conv_backward_x_op.h
python/jittor/extern/cuda/cudnn/ops/cudnn_conv_op.cc
python/jittor/extern/cuda/cudnn/ops/cudnn_conv_op.h
python/jittor/extern/cuda/cudnn/ops/cudnn_rnn_backward_x_op.cc
python/jittor/extern/cuda/cudnn/ops/cudnn_rnn_backward_x_op.h
python/jittor/extern/cuda/cudnn/ops/cudnn_rnn_op.cc
python/jittor/extern/cuda/cudnn/ops/cudnn_rnn_op.h
python/jittor/extern/cuda/cudnn/ops/cudnn_test_op.cc
python/jittor/extern/cuda/cudnn/ops/cudnn_test_op.h
python/jittor/extern/cuda/cudnn/src/cudnn_conv_test.cc
python/jittor/extern/cuda/cudnn/src/cudnn_rnn_descriptor.cc
python/jittor/extern/cuda/cudnn/src/cudnn_warper.cc
python/jittor/extern/cuda/cudnn/src/helper_cudnn.cc
python/jittor/extern/cuda/curand/inc/curand_warper.h
python/jittor/extern/cuda/curand/ops/curand_random_op.cc
python/jittor/extern/cuda/curand/ops/curand_random_op.h
python/jittor/extern/cuda/curand/src/curand_warper.cc
python/jittor/extern/cuda/curand/src/helper_curand.cc
python/jittor/extern/cuda/cutt/ops/cutt_test_op.cc
python/jittor/extern/cuda/cutt/ops/cutt_test_op.h
python/jittor/extern/cuda/cutt/ops/cutt_transpose_op.cc
python/jittor/extern/cuda/cutt/ops/cutt_transpose_op.h
python/jittor/extern/cuda/cutt/ops/cutt_warper.cc
python/jittor/extern/cuda/cutt/ops/cutt_warper.h
python/jittor/extern/cuda/inc/fp16_dev.h
python/jittor/extern/cuda/inc/fp16_emu.h
python/jittor/extern/cuda/inc/helper_cuda.h
python/jittor/extern/cuda/inc/helper_functions.h
python/jittor/extern/cuda/inc/helper_image.h
python/jittor/extern/cuda/inc/helper_string.h
python/jittor/extern/cuda/inc/helper_timer.h
python/jittor/extern/cuda/nccl/inc/nccl_warper.h
python/jittor/extern/cuda/nccl/ops/nccl_all_reduce_op.cc
python/jittor/extern/cuda/nccl/ops/nccl_all_reduce_op.h
python/jittor/extern/cuda/nccl/ops/nccl_broadcast_op.cc
python/jittor/extern/cuda/nccl/ops/nccl_broadcast_op.h
python/jittor/extern/cuda/nccl/ops/nccl_reduce_op.cc
python/jittor/extern/cuda/nccl/ops/nccl_reduce_op.h
python/jittor/extern/cuda/nccl/ops/nccl_test_op.cc
python/jittor/extern/cuda/nccl/ops/nccl_test_op.h
python/jittor/extern/cuda/nccl/src/nccl_warper.cc
python/jittor/extern/cuda/src/fp16_dev.cu
python/jittor/extern/cuda/src/fp16_emu.cpp
python/jittor/extern/cuda/src/helper_cuda.cc
python/jittor/extern/llvm/jt_alignment_from_assumptions.cc
python/jittor/extern/mkl/ops/cpu_cnn_inference_f32.cpp
python/jittor/extern/mkl/ops/mkl_conv_backward_w_op.cc
python/jittor/extern/mkl/ops/mkl_conv_backward_w_op.h
python/jittor/extern/mkl/ops/mkl_conv_backward_x_op.cc
python/jittor/extern/mkl/ops/mkl_conv_backward_x_op.h
python/jittor/extern/mkl/ops/mkl_conv_op.cc
python/jittor/extern/mkl/ops/mkl_conv_op.h
python/jittor/extern/mkl/ops/mkl_matmul_op.cc
python/jittor/extern/mkl/ops/mkl_matmul_op.h
python/jittor/extern/mkl/ops/mkl_test_op.cc
python/jittor/extern/mkl/ops/mkl_test_op.h
python/jittor/extern/mpi/inc/mpi_warper.h
python/jittor/extern/mpi/ops/mpi_all_reduce_op.cc
python/jittor/extern/mpi/ops/mpi_all_reduce_op.h
python/jittor/extern/mpi/ops/mpi_broadcast_op.cc
python/jittor/extern/mpi/ops/mpi_broadcast_op.h
python/jittor/extern/mpi/ops/mpi_reduce_op.cc
python/jittor/extern/mpi/ops/mpi_reduce_op.h
python/jittor/extern/mpi/ops/mpi_test_op.cc
python/jittor/extern/mpi/ops/mpi_test_op.h
python/jittor/extern/mpi/src/mpi_warper.cc
python/jittor/loss3d/__init__.py
python/jittor/loss3d/chamfer.py
python/jittor/loss3d/emd.py
python/jittor/models/__init__.py
python/jittor/models/alexnet.py
python/jittor/models/densenet.py
python/jittor/models/googlenet.py
python/jittor/models/inception.py
python/jittor/models/mnasnet.py
python/jittor/models/mobilenet.py
python/jittor/models/res2net.py
python/jittor/models/resnet.py
python/jittor/models/shufflenetv2.py
python/jittor/models/squeezenet.py
python/jittor/models/vgg.py
python/jittor/notebook/ConditionGAN.cn.md
python/jittor/notebook/ConditionGAN.md
python/jittor/notebook/ConditionGAN.src.md
python/jittor/notebook/LSGAN.cn.md
python/jittor/notebook/LSGAN.md
python/jittor/notebook/LSGAN.src.md
python/jittor/notebook/__main__.py
python/jittor/notebook/basics.cn.md
python/jittor/notebook/basics.md
python/jittor/notebook/basics.src.md
python/jittor/notebook/custom_op.cn.md
python/jittor/notebook/custom_op.md
python/jittor/notebook/custom_op.src.md
python/jittor/notebook/example.cn.md
python/jittor/notebook/example.md
python/jittor/notebook/example.src.md
python/jittor/notebook/md_to_ipynb.py
python/jittor/notebook/meta_op.cn.md
python/jittor/notebook/meta_op.md
python/jittor/notebook/meta_op.src.md
python/jittor/notebook/profiler.cn.md
python/jittor/notebook/profiler.md
python/jittor/notebook/profiler.src.md
python/jittor/notebook/60分钟快速入门Jittor/README.md
python/jittor/notebook/60分钟快速入门Jittor/mnist.png
python/jittor/notebook/60分钟快速入门Jittor/计图入门教程 0 --- 介绍与安装.ipynb
python/jittor/notebook/60分钟快速入门Jittor/计图入门教程 1 --- 基本概念.ipynb
python/jittor/notebook/60分钟快速入门Jittor/计图入门教程 2 --- 如何训练一个简单线性回归.ipynb
python/jittor/notebook/60分钟快速入门Jittor/计图入门教程 3 --- 尝试解决一个实际问题.ipynb
python/jittor/notebook/figs/mop.svg
python/jittor/script/Dockerfile_cuda11
python/jittor/script/build_aarch64_mkl.sh
python/jittor/script/converter_server.sh
python/jittor/script/inference_perf.py
python/jittor/script/install.sh
python/jittor/script/install_llvm.sh
python/jittor/script/install_mkl.sh
python/jittor/script/make_doc.py
python/jittor/script/tmpi
python/jittor/script/update.sh
python/jittor/src/common.h
python/jittor/src/core.h
python/jittor/src/event_queue.cc
python/jittor/src/event_queue.h
python/jittor/src/executor.cc
python/jittor/src/executor.h
python/jittor/src/fused_op.cc
python/jittor/src/fused_op.h
python/jittor/src/fuser.h
python/jittor/src/grad.cc
python/jittor/src/grad.h
python/jittor/src/graph.cc
python/jittor/src/graph.h
python/jittor/src/init.cc
python/jittor/src/init.h
python/jittor/src/jit_compiler.cc
python/jittor/src/jit_compiler.h
python/jittor/src/jit_key.cc
python/jittor/src/jit_key.h
python/jittor/src/lock.cc
python/jittor/src/lock.h
python/jittor/src/memory_profiler.cc
python/jittor/src/memory_profiler.h
python/jittor/src/node.h
python/jittor/src/numpy_func.h
python/jittor/src/op.cc
python/jittor/src/op.h
python/jittor/src/op_compiler.cc
python/jittor/src/op_compiler.h
python/jittor/src/parallel_compiler.cc
python/jittor/src/parallel_compiler.h
python/jittor/src/types.h
python/jittor/src/update_queue.cc
python/jittor/src/update_queue.h
python/jittor/src/var.cc
python/jittor/src/var.h
python/jittor/src/var_holder.cc
python/jittor/src/var_holder.h
python/jittor/src/var_slices.cc
python/jittor/src/var_slices.h
python/jittor/src/mem/allocator.cc
python/jittor/src/mem/allocator.h
python/jittor/src/mem/mem_info.cc
python/jittor/src/mem/mem_info.h
python/jittor/src/mem/allocator/aligned_allocator.cc
python/jittor/src/mem/allocator/aligned_allocator.h
python/jittor/src/mem/allocator/cuda_device_allocator.cc
python/jittor/src/mem/allocator/cuda_device_allocator.h
python/jittor/src/mem/allocator/cuda_dual_allocator.cc
python/jittor/src/mem/allocator/cuda_dual_allocator.h
python/jittor/src/mem/allocator/cuda_host_allocator.cc
python/jittor/src/mem/allocator/cuda_host_allocator.h
python/jittor/src/mem/allocator/cuda_managed_allocator.cc
python/jittor/src/mem/allocator/cuda_managed_allocator.h
python/jittor/src/mem/allocator/nfef_allocator.cc
python/jittor/src/mem/allocator/nfef_allocator.h
python/jittor/src/mem/allocator/sfrl_allocator.cc
python/jittor/src/mem/allocator/sfrl_allocator.h
python/jittor/src/mem/allocator/stat_allocator.cc
python/jittor/src/mem/allocator/stat_allocator.h
python/jittor/src/mem/allocator/temp_allocator.cc
python/jittor/src/mem/allocator/temp_allocator.h
python/jittor/src/misc/cpu_atomic.cc
python/jittor/src/misc/cpu_atomic.h
python/jittor/src/misc/cstr.h
python/jittor/src/misc/cuda_atomic.h
python/jittor/src/misc/cuda_flags.cc
python/jittor/src/misc/cuda_flags.h
python/jittor/src/misc/cuda_limits.h
python/jittor/src/misc/deleter.h
python/jittor/src/misc/fast_shared_ptr.h
python/jittor/src/misc/hash.h
python/jittor/src/misc/intrin.h
python/jittor/src/misc/nan_checker.cc
python/jittor/src/misc/nan_checker.cu
python/jittor/src/misc/nan_checker.h
python/jittor/src/misc/nano_string.cc
python/jittor/src/misc/nano_string.h
python/jittor/src/misc/nano_vector.h
python/jittor/src/misc/ring_buffer.cc
python/jittor/src/misc/ring_buffer.h
python/jittor/src/misc/stack_vector.h
python/jittor/src/misc/string_view_map.h
python/jittor/src/ops/arg_reduce_op.cc
python/jittor/src/ops/arg_reduce_op.h
python/jittor/src/ops/argsort_op.cc
python/jittor/src/ops/argsort_op.h
python/jittor/src/ops/array_op.cc
python/jittor/src/ops/array_op.h
python/jittor/src/ops/binary_op.cc
python/jittor/src/ops/binary_op.h
python/jittor/src/ops/binary_op_defs.h
python/jittor/src/ops/broadcast_to_op.cc
python/jittor/src/ops/broadcast_to_op.h
python/jittor/src/ops/candidate_op.cc
python/jittor/src/ops/candidate_op.h
python/jittor/src/ops/clone_op.cc
python/jittor/src/ops/clone_op.h
python/jittor/src/ops/code_op.cc
python/jittor/src/ops/code_op.h
python/jittor/src/ops/copy_op.cc
python/jittor/src/ops/copy_op.h
python/jittor/src/ops/empty_op.cc
python/jittor/src/ops/empty_op.h
python/jittor/src/ops/fetch_op.cc
python/jittor/src/ops/fetch_op.h
python/jittor/src/ops/fuse_transpose_op.cc
python/jittor/src/ops/fuse_transpose_op.h
python/jittor/src/ops/getitem_op.cc
python/jittor/src/ops/getitem_op.h
python/jittor/src/ops/index_op.cc
python/jittor/src/ops/index_op.h
python/jittor/src/ops/numpy_code_op.cc
python/jittor/src/ops/numpy_code_op.h
python/jittor/src/ops/op_register.cc
python/jittor/src/ops/op_register.h
python/jittor/src/ops/op_utils.cc
python/jittor/src/ops/random_op.cc
python/jittor/src/ops/random_op.h
python/jittor/src/ops/reduce_op.cc
python/jittor/src/ops/reduce_op.h
python/jittor/src/ops/reindex_op.cc
python/jittor/src/ops/reindex_op.h
python/jittor/src/ops/reindex_reduce_op.cc
python/jittor/src/ops/reindex_reduce_op.h
python/jittor/src/ops/reshape_op.cc
python/jittor/src/ops/reshape_op.h
python/jittor/src/ops/safe_clip_op.cc
python/jittor/src/ops/safe_clip_op.h
python/jittor/src/ops/setitem_op.cc
python/jittor/src/ops/setitem_op.h
python/jittor/src/ops/tape_op.cc
python/jittor/src/ops/tape_op.h
python/jittor/src/ops/ternary_op.cc
python/jittor/src/ops/ternary_op.h
python/jittor/src/ops/transpose_op.cc
python/jittor/src/ops/transpose_op.h
python/jittor/src/ops/unary_op.cc
python/jittor/src/ops/unary_op.h
python/jittor/src/ops/unary_op_defs.h
python/jittor/src/ops/where_op.cc
python/jittor/src/ops/where_op.h
python/jittor/src/opt/expr.cc
python/jittor/src/opt/expr.h
python/jittor/src/opt/jit_searcher.cc
python/jittor/src/opt/jit_searcher.h
python/jittor/src/opt/kernel_ir.cc
python/jittor/src/opt/kernel_ir.h
python/jittor/src/opt/pass_manager.cc
python/jittor/src/opt/pass_manager.h
python/jittor/src/opt/tuner_manager.cc
python/jittor/src/opt/tuner_manager.h
python/jittor/src/opt/var_relay.cc
python/jittor/src/opt/var_relay.h
python/jittor/src/opt/gopt/setitem_gopt.cc
python/jittor/src/opt/pass/assume_aligned_pass.cc
python/jittor/src/opt/pass/assume_aligned_pass.h
python/jittor/src/opt/pass/atomic_tuner_pass.h
python/jittor/src/opt/pass/check_cache_pass.cc
python/jittor/src/opt/pass/check_cache_pass.h
python/jittor/src/opt/pass/compile_shapes_pass.cc
python/jittor/src/opt/pass/compile_shapes_pass.h
python/jittor/src/opt/pass/const_var_pass.cc
python/jittor/src/opt/pass/const_var_pass.h
python/jittor/src/opt/pass/expand_empty_block_pass.cc
python/jittor/src/opt/pass/expand_empty_block_pass.h
python/jittor/src/opt/pass/fake_main_pass.cc
python/jittor/src/opt/pass/fake_main_pass.h
python/jittor/src/opt/pass/float_atomic_fix_pass.cc
python/jittor/src/opt/pass/float_atomic_fix_pass.h
python/jittor/src/opt/pass/insert_profile_loop_pass.cc
python/jittor/src/opt/pass/insert_profile_loop_pass.h
python/jittor/src/opt/pass/loop_to_func_pass.cc
python/jittor/src/opt/pass/loop_to_func_pass.h
python/jittor/src/opt/pass/loop_var_analyze_pass.cc
python/jittor/src/opt/pass/loop_var_analyze_pass.h
python/jittor/src/opt/pass/mark_raw_pass.cc
python/jittor/src/opt/pass/mark_raw_pass.h
python/jittor/src/opt/pass/merge_loop_pass.cc
python/jittor/src/opt/pass/merge_loop_pass.h
python/jittor/src/opt/pass/merge_loop_var_pass.cc
python/jittor/src/opt/pass/merge_loop_var_pass.h
python/jittor/src/opt/pass/parallel_pass.h
python/jittor/src/opt/pass/pass.cc
python/jittor/src/opt/pass/pass.h
python/jittor/src/opt/pass/remove_intermediate_pass.cc
python/jittor/src/opt/pass/remove_intermediate_pass.h
python/jittor/src/opt/pass/remove_loop_pass.cc
python/jittor/src/opt/pass/remove_loop_pass.h
python/jittor/src/opt/pass/rename_loop_index_pass.cc
python/jittor/src/opt/pass/rename_loop_index_pass.h
python/jittor/src/opt/pass/reorder_loop_pass.cc
python/jittor/src/opt/pass/reorder_loop_pass.h
python/jittor/src/opt/pass/replace_for_num_pass.cc
python/jittor/src/opt/pass/replace_for_num_pass.h
python/jittor/src/opt/pass/restride_pass.cc
python/jittor/src/opt/pass/restride_pass.h
python/jittor/src/opt/pass/solve_conflict_define_pass.cc
python/jittor/src/opt/pass/solve_conflict_define_pass.h
python/jittor/src/opt/pass/split_loop_pass.cc
python/jittor/src/opt/pass/split_loop_pass.h
python/jittor/src/opt/pass/unroll_pass.cc
python/jittor/src/opt/pass/unroll_pass.h
python/jittor/src/opt/pass/use_movnt_pass.cc
python/jittor/src/opt/pass/use_movnt_pass.h
python/jittor/src/opt/pass/vectorize_pass.cc
python/jittor/src/opt/pass/vectorize_pass.h
python/jittor/src/opt/tuner/broadcast_tuner.cc
python/jittor/src/opt/tuner/broadcast_tuner.h
python/jittor/src/opt/tuner/conv_tuner.cc
python/jittor/src/opt/tuner/conv_tuner.h
python/jittor/src/opt/tuner/matmul_tuner.cc
python/jittor/src/opt/tuner/matmul_tuner.h
python/jittor/src/opt/tuner/reduce_tuner.cc
python/jittor/src/opt/tuner/reduce_tuner.h
python/jittor/src/opt/tuner/reorder_tuner.cc
python/jittor/src/opt/tuner/reorder_tuner.h
python/jittor/src/opt/tuner/tuner.cc
python/jittor/src/opt/tuner/tuner.h
python/jittor/src/profiler/cache_info.cc
python/jittor/src/profiler/cache_info.h
python/jittor/src/profiler/memory_checker.cc
python/jittor/src/profiler/memory_checker.h
python/jittor/src/profiler/profiler.cc
python/jittor/src/profiler/profiler.h
python/jittor/src/profiler/profiler_guard.h
python/jittor/src/profiler/replacement.cc
python/jittor/src/profiler/replacement.h
python/jittor/src/profiler/simple_profiler.h
python/jittor/src/profiler/vtop.cc
python/jittor/src/pybind/core.cc
python/jittor/src/pybind/py_var_tracer.cc
python/jittor/src/pybind/py_var_tracer.h
python/jittor/src/pybind/py_var_tracer_interface.h
python/jittor/src/pyjt/numpy.cc
python/jittor/src/pyjt/numpy.h
python/jittor/src/pyjt/py_arg_printer.cc
python/jittor/src/pyjt/py_arg_printer.h
python/jittor/src/pyjt/py_array_op.cc
python/jittor/src/pyjt/py_caller.cc
python/jittor/src/pyjt/py_caller.h
python/jittor/src/pyjt/py_converter.h
python/jittor/src/pyjt/py_obj_holder.h
python/jittor/src/pyjt/py_ring_buffer.cc
python/jittor/src/pyjt/py_ring_buffer.h
python/jittor/src/pyjt/pyjt_console.h
python/jittor/src/test/test_expr.cc
python/jittor/src/test/test_fast_shared_ptr.cc
python/jittor/src/test/test_jit_key.cc
python/jittor/src/test/test_kernel_ir.cc
python/jittor/src/test/test_nano_vector.cc
python/jittor/src/test/test_op_compiler.cc
python/jittor/src/test/test_op_relay.cc
python/jittor/src/test/test_setitem_op.cc
python/jittor/src/test/test_sfrl_allocator.cc
python/jittor/src/utils/cache_compile.cc
python/jittor/src/utils/cache_compile.h
python/jittor/src/utils/cross_platform.h
python/jittor/src/utils/flags.cc
python/jittor/src/utils/flags.h
python/jittor/src/utils/jit_utils.cc
python/jittor/src/utils/log.cc
python/jittor/src/utils/log.h
python/jittor/src/utils/mwsr_list.cc
python/jittor/src/utils/mwsr_list.h
python/jittor/src/utils/seh.h
python/jittor/src/utils/str_utils.cc
python/jittor/src/utils/str_utils.h
python/jittor/src/utils/tracer.cc
python/jittor/src/utils/tracer.h
python/jittor/src/utils/vdp
python/jittor/test/__main__.py
python/jittor/test/test.h
python/jittor/test/test_adamw.py
python/jittor/test/test_affine_grid.py
python/jittor/test/test_allocator.py
python/jittor/test/test_allocator2.py
python/jittor/test/test_arg_pool_op.py
python/jittor/test/test_arg_reduce_op.py
python/jittor/test/test_argsort_op.py
python/jittor/test/test_array.py
python/jittor/test/test_asm_tuner.py
python/jittor/test/test_atomic_tuner.py
python/jittor/test/test_attention.py
python/jittor/test/test_auto_diff.py
python/jittor/test/test_batchnorm.py
python/jittor/test/test_bicubic.py
python/jittor/test/test_binary_op.py
python/jittor/test/test_bmm.py
python/jittor/test/test_broadcast_to_op.py
python/jittor/test/test_broadcast_tuner.py
python/jittor/test/test_cache.py
python/jittor/test/test_candidate.py
python/jittor/test/test_clone.py
python/jittor/test/test_code_op.py
python/jittor/test/test_compile_options.py
python/jittor/test/test_concat_op.py
python/jittor/test/test_console.py
python/jittor/test/test_contrib.py
python/jittor/test/test_conv_transpose.py
python/jittor/test/test_conv_tuner.py
python/jittor/test/test_core.py
python/jittor/test/test_cub_cumsum.py
python/jittor/test/test_cublas_test_op.py
python/jittor/test/test_cuda.py
python/jittor/test/test_cudnn_op.py
python/jittor/test/test_cumprod_op.py
python/jittor/test/test_custom_op.py
python/jittor/test/test_cutt.py
python/jittor/test/test_cutt_transpose_op.py
python/jittor/test/test_dataset.py
python/jittor/test/test_default_var.py
python/jittor/test/test_densenet.py
python/jittor/test/test_depthwise_conv.py
python/jittor/test/test_distributions.py
python/jittor/test/test_emnist.py
python/jittor/test/test_example.py
python/jittor/test/test_example_accumulate_grad.py
python/jittor/test/test_fetcher.py
python/jittor/test/test_flags.py
python/jittor/test/test_fold.py
python/jittor/test/test_function.py
python/jittor/test/test_fused_op.py
python/jittor/test/test_fuser.py
python/jittor/test/test_grad.py
python/jittor/test/test_group_conv_tuner.py
python/jittor/test/test_hook.py
python/jittor/test/test_image_folder.py
python/jittor/test/test_inception.py
python/jittor/test/test_index_op.py
python/jittor/test/test_init.py
python/jittor/test/test_jit_tests.py
python/jittor/test/test_jtune.py
python/jittor/test/test_lazy_execution.py
python/jittor/test/test_linalg.py
python/jittor/test/test_load_pth.py
python/jittor/test/test_lock.py
python/jittor/test/test_log.py
python/jittor/test/test_longest_dis_fuse.py
python/jittor/test/test_loss.py
python/jittor/test/test_loss3d.py
python/jittor/test/test_lr_scheduler.py
python/jittor/test/test_lstm.py
python/jittor/test/test_matmul.py
python/jittor/test/test_matmul_tuner.py
python/jittor/test/test_mem.py
python/jittor/test/test_memory_profiler.py
python/jittor/test/test_merge_loop_var_pass.py
python/jittor/test/test_merge_single_array_op.py
python/jittor/test/test_misc_issue.py
python/jittor/test/test_misc_op.py
python/jittor/test/test_mkl_conv_op.py
python/jittor/test/test_mkl_test_op.py
python/jittor/test/test_models.py
python/jittor/test/test_mpi.py
python/jittor/test/test_mpi_batchnorm.py
python/jittor/test/test_mpi_op.py
python/jittor/test/test_nano_string.py
python/jittor/test/test_nano_vector.py
python/jittor/test/test_nccl.py
python/jittor/test/test_nccl_ops.py
python/jittor/test/test_new_fused_op.py
python/jittor/test/test_node.py
python/jittor/test/test_notebooks.py
python/jittor/test/test_numpy_code_op.py
python/jittor/test/test_op_compiler.py
python/jittor/test/test_opt_state_dict.py
python/jittor/test/test_optimizer.py
python/jittor/test/test_pad.py
python/jittor/test/test_parallel_pass.py
python/jittor/test/test_param_list.py
python/jittor/test/test_pytorch_converter.py
python/jittor/test/test_random_op.py
python/jittor/test/test_reduce_op.py
python/jittor/test/test_reduce_tuner.py
python/jittor/test/test_reindex_op.py
python/jittor/test/test_reindex_reduce_op.py
python/jittor/test/test_relu.py
python/jittor/test/test_reorder_tuner.py
python/jittor/test/test_repeat.py
python/jittor/test/test_reshape.py
python/jittor/test/test_resize_and_crop.py
python/jittor/test/test_resnet.py
python/jittor/test/test_ring_buffer.py
python/jittor/test/test_ring_buffer2.py
python/jittor/test/test_rnn.py
python/jittor/test/test_sampler.py
python/jittor/test/test_search_sorted.py
python/jittor/test/test_searchsorted_op.py
python/jittor/test/test_setitem.py
python/jittor/test/test_single_process_scope.py
python/jittor/test/test_slice.py
python/jittor/test/test_sparse.py
python/jittor/test/test_std.py
python/jittor/test/test_stop_fuse.py
python/jittor/test/test_ternary_op.py
python/jittor/test/test_trace_var.py
python/jittor/test/test_tracer.py
python/jittor/test/test_transform.py
python/jittor/test/test_transpose_op.py
python/jittor/test/test_unary_op.py
python/jittor/test/test_utils.py
python/jittor/test/test_vgg.py
python/jittor/test/test_where_op.py
python/jittor/test/perf/perf.py
python/jittor/test/system/test_all.sh
python/jittor/test/system/test_cuda10.0_ubuntu16.04.sh
python/jittor/test/system/test_cuda10.0_ubuntu18.04.sh
python/jittor/test/system/test_cuda11.1_ubuntu16.04.sh
python/jittor/test/system/test_cuda11.1_ubuntu18.04.sh
python/jittor/test/system/test_cuda11.1_ubuntu20.04.sh
python/jittor/test/system/test_nocuda_ubuntu18.04.sh
python/jittor/transform/__init__.py
python/jittor/transform/function_pil.py
python/jittor/utils/asm_tuner.py
python/jittor/utils/autocomplete.py
python/jittor/utils/code_trans.py
python/jittor/utils/converter_server.py
python/jittor/utils/data.gz
python/jittor/utils/dumpdef.py
python/jittor/utils/jtune.py
python/jittor/utils/local_doc_builder.py
python/jittor/utils/nvtx.py
python/jittor/utils/polish.py
python/jittor/utils/polish_centos.py
python/jittor/utils/publish.py
python/jittor/utils/pytorch_converter.py
python/jittor/utils/tracer.py
python/jittor_utils/__init__.py
python/jittor_utils/auto_diff.py
python/jittor_utils/clean_cache.py
python/jittor_utils/config.py
python/jittor_utils/install_cuda.py
python/jittor_utils/install_msvc.py
python/jittor_utils/lock.py
python/jittor_utils/misc.py
python/jittor_utils/pip_publish.py
python/jittor_utils/ring_buffer.py
python/jittor_utils/student_queue.py
python/jittor_utils/translator.py
python/jittor_utils/class/motd
python/jittor_utils/class/setup.py
python/jittor_utils/class/setup_env.py