diff --git a/extern/cuda/cub/inc/cub_test.h b/python/jittor/extern/cuda/cub/inc/cub_test.h
similarity index 100%
rename from extern/cuda/cub/inc/cub_test.h
rename to python/jittor/extern/cuda/cub/inc/cub_test.h
diff --git a/extern/cuda/cub/ops/cub_arg_reduce_op.cc b/python/jittor/extern/cuda/cub/ops/cub_arg_reduce_op.cc
similarity index 100%
rename from extern/cuda/cub/ops/cub_arg_reduce_op.cc
rename to python/jittor/extern/cuda/cub/ops/cub_arg_reduce_op.cc
diff --git a/extern/cuda/cub/ops/cub_arg_reduce_op.h b/python/jittor/extern/cuda/cub/ops/cub_arg_reduce_op.h
similarity index 100%
rename from extern/cuda/cub/ops/cub_arg_reduce_op.h
rename to python/jittor/extern/cuda/cub/ops/cub_arg_reduce_op.h
diff --git a/extern/cuda/cub/ops/cub_argsort_op.cc b/python/jittor/extern/cuda/cub/ops/cub_argsort_op.cc
similarity index 100%
rename from extern/cuda/cub/ops/cub_argsort_op.cc
rename to python/jittor/extern/cuda/cub/ops/cub_argsort_op.cc
diff --git a/extern/cuda/cub/ops/cub_argsort_op.h b/python/jittor/extern/cuda/cub/ops/cub_argsort_op.h
similarity index 100%
rename from extern/cuda/cub/ops/cub_argsort_op.h
rename to python/jittor/extern/cuda/cub/ops/cub_argsort_op.h
diff --git a/extern/cuda/cub/ops/cub_test_op.cc b/python/jittor/extern/cuda/cub/ops/cub_test_op.cc
similarity index 100%
rename from extern/cuda/cub/ops/cub_test_op.cc
rename to python/jittor/extern/cuda/cub/ops/cub_test_op.cc
diff --git a/extern/cuda/cub/ops/cub_test_op.h b/python/jittor/extern/cuda/cub/ops/cub_test_op.h
similarity index 100%
rename from extern/cuda/cub/ops/cub_test_op.h
rename to python/jittor/extern/cuda/cub/ops/cub_test_op.h
diff --git a/extern/cuda/cub/ops/cub_where_op.cc b/python/jittor/extern/cuda/cub/ops/cub_where_op.cc
similarity index 100%
rename from extern/cuda/cub/ops/cub_where_op.cc
rename to python/jittor/extern/cuda/cub/ops/cub_where_op.cc
diff --git a/extern/cuda/cub/ops/cub_where_op.h b/python/jittor/extern/cuda/cub/ops/cub_where_op.h
similarity index 100%
rename from extern/cuda/cub/ops/cub_where_op.h
rename to python/jittor/extern/cuda/cub/ops/cub_where_op.h
diff --git a/extern/cuda/cublas/inc/cublas_warper.h b/python/jittor/extern/cuda/cublas/inc/cublas_warper.h
similarity index 100%
rename from extern/cuda/cublas/inc/cublas_warper.h
rename to python/jittor/extern/cuda/cublas/inc/cublas_warper.h
diff --git a/extern/cuda/cublas/ops/cublas_batched_matmul_op.cc b/python/jittor/extern/cuda/cublas/ops/cublas_batched_matmul_op.cc
similarity index 100%
rename from extern/cuda/cublas/ops/cublas_batched_matmul_op.cc
rename to python/jittor/extern/cuda/cublas/ops/cublas_batched_matmul_op.cc
diff --git a/extern/cuda/cublas/ops/cublas_batched_matmul_op.h b/python/jittor/extern/cuda/cublas/ops/cublas_batched_matmul_op.h
similarity index 100%
rename from extern/cuda/cublas/ops/cublas_batched_matmul_op.h
rename to python/jittor/extern/cuda/cublas/ops/cublas_batched_matmul_op.h
diff --git a/extern/cuda/cublas/ops/cublas_matmul_op.cc b/python/jittor/extern/cuda/cublas/ops/cublas_matmul_op.cc
similarity index 100%
rename from extern/cuda/cublas/ops/cublas_matmul_op.cc
rename to python/jittor/extern/cuda/cublas/ops/cublas_matmul_op.cc
diff --git a/extern/cuda/cublas/ops/cublas_matmul_op.h b/python/jittor/extern/cuda/cublas/ops/cublas_matmul_op.h
similarity index 100%
rename from extern/cuda/cublas/ops/cublas_matmul_op.h
rename to python/jittor/extern/cuda/cublas/ops/cublas_matmul_op.h
diff --git a/extern/cuda/cublas/ops/cublas_test_op.cc b/python/jittor/extern/cuda/cublas/ops/cublas_test_op.cc
similarity index 100%
rename from extern/cuda/cublas/ops/cublas_test_op.cc
rename to python/jittor/extern/cuda/cublas/ops/cublas_test_op.cc
diff --git a/extern/cuda/cublas/ops/cublas_test_op.h b/python/jittor/extern/cuda/cublas/ops/cublas_test_op.h
similarity index 100%
rename from extern/cuda/cublas/ops/cublas_test_op.h
rename to python/jittor/extern/cuda/cublas/ops/cublas_test_op.h
diff --git a/extern/cuda/cublas/src/cublas_matmul_test.cc b/python/jittor/extern/cuda/cublas/src/cublas_matmul_test.cc
similarity index 100%
rename from extern/cuda/cublas/src/cublas_matmul_test.cc
rename to python/jittor/extern/cuda/cublas/src/cublas_matmul_test.cc
diff --git a/extern/cuda/cublas/src/cublas_warper.cc b/python/jittor/extern/cuda/cublas/src/cublas_warper.cc
similarity index 100%
rename from extern/cuda/cublas/src/cublas_warper.cc
rename to python/jittor/extern/cuda/cublas/src/cublas_warper.cc
diff --git a/extern/cuda/cublas/src/helper_cublas.cc b/python/jittor/extern/cuda/cublas/src/helper_cublas.cc
similarity index 100%
rename from extern/cuda/cublas/src/helper_cublas.cc
rename to python/jittor/extern/cuda/cublas/src/helper_cublas.cc
diff --git a/extern/cuda/cudnn/inc/cudnn_warper.h b/python/jittor/extern/cuda/cudnn/inc/cudnn_warper.h
similarity index 100%
rename from extern/cuda/cudnn/inc/cudnn_warper.h
rename to python/jittor/extern/cuda/cudnn/inc/cudnn_warper.h
diff --git a/extern/cuda/cudnn/ops/cudnn_conv_backward_w_op.cc b/python/jittor/extern/cuda/cudnn/ops/cudnn_conv_backward_w_op.cc
similarity index 100%
rename from extern/cuda/cudnn/ops/cudnn_conv_backward_w_op.cc
rename to python/jittor/extern/cuda/cudnn/ops/cudnn_conv_backward_w_op.cc
diff --git a/extern/cuda/cudnn/ops/cudnn_conv_backward_w_op.h b/python/jittor/extern/cuda/cudnn/ops/cudnn_conv_backward_w_op.h
similarity index 100%
rename from extern/cuda/cudnn/ops/cudnn_conv_backward_w_op.h
rename to python/jittor/extern/cuda/cudnn/ops/cudnn_conv_backward_w_op.h
diff --git a/extern/cuda/cudnn/ops/cudnn_conv_backward_x_op.cc b/python/jittor/extern/cuda/cudnn/ops/cudnn_conv_backward_x_op.cc
similarity index 100%
rename from extern/cuda/cudnn/ops/cudnn_conv_backward_x_op.cc
rename to python/jittor/extern/cuda/cudnn/ops/cudnn_conv_backward_x_op.cc
diff --git a/extern/cuda/cudnn/ops/cudnn_conv_backward_x_op.h b/python/jittor/extern/cuda/cudnn/ops/cudnn_conv_backward_x_op.h
similarity index 100%
rename from extern/cuda/cudnn/ops/cudnn_conv_backward_x_op.h
rename to python/jittor/extern/cuda/cudnn/ops/cudnn_conv_backward_x_op.h
diff --git a/extern/cuda/cudnn/ops/cudnn_conv_op.cc b/python/jittor/extern/cuda/cudnn/ops/cudnn_conv_op.cc
similarity index 100%
rename from extern/cuda/cudnn/ops/cudnn_conv_op.cc
rename to python/jittor/extern/cuda/cudnn/ops/cudnn_conv_op.cc
diff --git a/extern/cuda/cudnn/ops/cudnn_conv_op.h b/python/jittor/extern/cuda/cudnn/ops/cudnn_conv_op.h
similarity index 100%
rename from extern/cuda/cudnn/ops/cudnn_conv_op.h
rename to python/jittor/extern/cuda/cudnn/ops/cudnn_conv_op.h
diff --git a/extern/cuda/cudnn/ops/cudnn_test_op.cc b/python/jittor/extern/cuda/cudnn/ops/cudnn_test_op.cc
similarity index 100%
rename from extern/cuda/cudnn/ops/cudnn_test_op.cc
rename to python/jittor/extern/cuda/cudnn/ops/cudnn_test_op.cc
diff --git a/extern/cuda/cudnn/ops/cudnn_test_op.h b/python/jittor/extern/cuda/cudnn/ops/cudnn_test_op.h
similarity index 100%
rename from extern/cuda/cudnn/ops/cudnn_test_op.h
rename to python/jittor/extern/cuda/cudnn/ops/cudnn_test_op.h
diff --git a/extern/cuda/cudnn/src/cudnn_conv_test.cc b/python/jittor/extern/cuda/cudnn/src/cudnn_conv_test.cc
similarity index 100%
rename from extern/cuda/cudnn/src/cudnn_conv_test.cc
rename to python/jittor/extern/cuda/cudnn/src/cudnn_conv_test.cc
diff --git a/extern/cuda/cudnn/src/cudnn_warper.cc b/python/jittor/extern/cuda/cudnn/src/cudnn_warper.cc
similarity index 100%
rename from extern/cuda/cudnn/src/cudnn_warper.cc
rename to python/jittor/extern/cuda/cudnn/src/cudnn_warper.cc
diff --git a/extern/cuda/cudnn/src/helper_cudnn.cc b/python/jittor/extern/cuda/cudnn/src/helper_cudnn.cc
similarity index 100%
rename from extern/cuda/cudnn/src/helper_cudnn.cc
rename to python/jittor/extern/cuda/cudnn/src/helper_cudnn.cc
diff --git a/extern/cuda/curand/inc/curand_warper.h b/python/jittor/extern/cuda/curand/inc/curand_warper.h
similarity index 100%
rename from extern/cuda/curand/inc/curand_warper.h
rename to python/jittor/extern/cuda/curand/inc/curand_warper.h
diff --git a/extern/cuda/curand/ops/curand_random_op.cc b/python/jittor/extern/cuda/curand/ops/curand_random_op.cc
similarity index 100%
rename from extern/cuda/curand/ops/curand_random_op.cc
rename to python/jittor/extern/cuda/curand/ops/curand_random_op.cc
diff --git a/extern/cuda/curand/ops/curand_random_op.h b/python/jittor/extern/cuda/curand/ops/curand_random_op.h
similarity index 100%
rename from extern/cuda/curand/ops/curand_random_op.h
rename to python/jittor/extern/cuda/curand/ops/curand_random_op.h
diff --git a/extern/cuda/curand/src/curand_warper.cc b/python/jittor/extern/cuda/curand/src/curand_warper.cc
similarity index 100%
rename from extern/cuda/curand/src/curand_warper.cc
rename to python/jittor/extern/cuda/curand/src/curand_warper.cc
diff --git a/extern/cuda/curand/src/helper_curand.cc b/python/jittor/extern/cuda/curand/src/helper_curand.cc
similarity index 100%
rename from extern/cuda/curand/src/helper_curand.cc
rename to python/jittor/extern/cuda/curand/src/helper_curand.cc
diff --git a/extern/cuda/cutt/ops/cutt_test_op.cc b/python/jittor/extern/cuda/cutt/ops/cutt_test_op.cc
similarity index 100%
rename from extern/cuda/cutt/ops/cutt_test_op.cc
rename to python/jittor/extern/cuda/cutt/ops/cutt_test_op.cc
diff --git a/extern/cuda/cutt/ops/cutt_test_op.h b/python/jittor/extern/cuda/cutt/ops/cutt_test_op.h
similarity index 100%
rename from extern/cuda/cutt/ops/cutt_test_op.h
rename to python/jittor/extern/cuda/cutt/ops/cutt_test_op.h
diff --git a/extern/cuda/cutt/ops/cutt_transpose_op.cc b/python/jittor/extern/cuda/cutt/ops/cutt_transpose_op.cc
similarity index 100%
rename from extern/cuda/cutt/ops/cutt_transpose_op.cc
rename to python/jittor/extern/cuda/cutt/ops/cutt_transpose_op.cc
diff --git a/extern/cuda/cutt/ops/cutt_transpose_op.h b/python/jittor/extern/cuda/cutt/ops/cutt_transpose_op.h
similarity index 100%
rename from extern/cuda/cutt/ops/cutt_transpose_op.h
rename to python/jittor/extern/cuda/cutt/ops/cutt_transpose_op.h
diff --git a/extern/cuda/cutt/ops/cutt_warper.cc b/python/jittor/extern/cuda/cutt/ops/cutt_warper.cc
similarity index 100%
rename from extern/cuda/cutt/ops/cutt_warper.cc
rename to python/jittor/extern/cuda/cutt/ops/cutt_warper.cc
diff --git a/extern/cuda/cutt/ops/cutt_warper.h b/python/jittor/extern/cuda/cutt/ops/cutt_warper.h
similarity index 100%
rename from extern/cuda/cutt/ops/cutt_warper.h
rename to python/jittor/extern/cuda/cutt/ops/cutt_warper.h
diff --git a/extern/cuda/inc/fp16_dev.h b/python/jittor/extern/cuda/inc/fp16_dev.h
similarity index 100%
rename from extern/cuda/inc/fp16_dev.h
rename to python/jittor/extern/cuda/inc/fp16_dev.h
diff --git a/extern/cuda/inc/fp16_emu.h b/python/jittor/extern/cuda/inc/fp16_emu.h
similarity index 100%
rename from extern/cuda/inc/fp16_emu.h
rename to python/jittor/extern/cuda/inc/fp16_emu.h
diff --git a/extern/cuda/inc/helper_cuda.h b/python/jittor/extern/cuda/inc/helper_cuda.h
similarity index 100%
rename from extern/cuda/inc/helper_cuda.h
rename to python/jittor/extern/cuda/inc/helper_cuda.h
diff --git a/extern/cuda/inc/helper_functions.h b/python/jittor/extern/cuda/inc/helper_functions.h
similarity index 100%
rename from extern/cuda/inc/helper_functions.h
rename to python/jittor/extern/cuda/inc/helper_functions.h
diff --git a/extern/cuda/inc/helper_image.h b/python/jittor/extern/cuda/inc/helper_image.h
similarity index 100%
rename from extern/cuda/inc/helper_image.h
rename to python/jittor/extern/cuda/inc/helper_image.h
diff --git a/extern/cuda/inc/helper_string.h b/python/jittor/extern/cuda/inc/helper_string.h
similarity index 100%
rename from extern/cuda/inc/helper_string.h
rename to python/jittor/extern/cuda/inc/helper_string.h
diff --git a/extern/cuda/inc/helper_timer.h b/python/jittor/extern/cuda/inc/helper_timer.h
similarity index 100%
rename from extern/cuda/inc/helper_timer.h
rename to python/jittor/extern/cuda/inc/helper_timer.h
diff --git a/extern/cuda/nccl/inc/nccl_warper.h b/python/jittor/extern/cuda/nccl/inc/nccl_warper.h
similarity index 100%
rename from extern/cuda/nccl/inc/nccl_warper.h
rename to python/jittor/extern/cuda/nccl/inc/nccl_warper.h
diff --git a/extern/cuda/nccl/ops/nccl_all_reduce_op.cc b/python/jittor/extern/cuda/nccl/ops/nccl_all_reduce_op.cc
similarity index 100%
rename from extern/cuda/nccl/ops/nccl_all_reduce_op.cc
rename to python/jittor/extern/cuda/nccl/ops/nccl_all_reduce_op.cc
diff --git a/extern/cuda/nccl/ops/nccl_all_reduce_op.h b/python/jittor/extern/cuda/nccl/ops/nccl_all_reduce_op.h
similarity index 100%
rename from extern/cuda/nccl/ops/nccl_all_reduce_op.h
rename to python/jittor/extern/cuda/nccl/ops/nccl_all_reduce_op.h
diff --git a/extern/cuda/nccl/ops/nccl_broadcast_op.cc b/python/jittor/extern/cuda/nccl/ops/nccl_broadcast_op.cc
similarity index 100%
rename from extern/cuda/nccl/ops/nccl_broadcast_op.cc
rename to python/jittor/extern/cuda/nccl/ops/nccl_broadcast_op.cc
diff --git a/extern/cuda/nccl/ops/nccl_broadcast_op.h b/python/jittor/extern/cuda/nccl/ops/nccl_broadcast_op.h
similarity index 100%
rename from extern/cuda/nccl/ops/nccl_broadcast_op.h
rename to python/jittor/extern/cuda/nccl/ops/nccl_broadcast_op.h
diff --git a/extern/cuda/nccl/ops/nccl_reduce_op.cc b/python/jittor/extern/cuda/nccl/ops/nccl_reduce_op.cc
similarity index 100%
rename from extern/cuda/nccl/ops/nccl_reduce_op.cc
rename to python/jittor/extern/cuda/nccl/ops/nccl_reduce_op.cc
diff --git a/extern/cuda/nccl/ops/nccl_reduce_op.h b/python/jittor/extern/cuda/nccl/ops/nccl_reduce_op.h
similarity index 100%
rename from extern/cuda/nccl/ops/nccl_reduce_op.h
rename to python/jittor/extern/cuda/nccl/ops/nccl_reduce_op.h
diff --git a/extern/cuda/nccl/ops/nccl_test_op.cc b/python/jittor/extern/cuda/nccl/ops/nccl_test_op.cc
similarity index 100%
rename from extern/cuda/nccl/ops/nccl_test_op.cc
rename to python/jittor/extern/cuda/nccl/ops/nccl_test_op.cc
diff --git a/extern/cuda/nccl/ops/nccl_test_op.h b/python/jittor/extern/cuda/nccl/ops/nccl_test_op.h
similarity index 100%
rename from extern/cuda/nccl/ops/nccl_test_op.h
rename to python/jittor/extern/cuda/nccl/ops/nccl_test_op.h
diff --git a/extern/cuda/nccl/src/nccl_warper.cc b/python/jittor/extern/cuda/nccl/src/nccl_warper.cc
similarity index 100%
rename from extern/cuda/nccl/src/nccl_warper.cc
rename to python/jittor/extern/cuda/nccl/src/nccl_warper.cc
diff --git a/extern/cuda/src/fp16_dev.cu b/python/jittor/extern/cuda/src/fp16_dev.cu
similarity index 100%
rename from extern/cuda/src/fp16_dev.cu
rename to python/jittor/extern/cuda/src/fp16_dev.cu
diff --git a/extern/cuda/src/fp16_emu.cpp b/python/jittor/extern/cuda/src/fp16_emu.cpp
similarity index 100%
rename from extern/cuda/src/fp16_emu.cpp
rename to python/jittor/extern/cuda/src/fp16_emu.cpp
diff --git a/extern/cuda/src/helper_cuda.cc b/python/jittor/extern/cuda/src/helper_cuda.cc
similarity index 100%
rename from extern/cuda/src/helper_cuda.cc
rename to python/jittor/extern/cuda/src/helper_cuda.cc
diff --git a/extern/llvm/jt_alignment_from_assumptions.cc b/python/jittor/extern/llvm/jt_alignment_from_assumptions.cc
similarity index 100%
rename from extern/llvm/jt_alignment_from_assumptions.cc
rename to python/jittor/extern/llvm/jt_alignment_from_assumptions.cc
diff --git a/extern/mkl/ops/cpu_cnn_inference_f32.cpp b/python/jittor/extern/mkl/ops/cpu_cnn_inference_f32.cpp
similarity index 100%
rename from extern/mkl/ops/cpu_cnn_inference_f32.cpp
rename to python/jittor/extern/mkl/ops/cpu_cnn_inference_f32.cpp
diff --git a/extern/mkl/ops/mkl_conv_backward_w_op.cc b/python/jittor/extern/mkl/ops/mkl_conv_backward_w_op.cc
similarity index 100%
rename from extern/mkl/ops/mkl_conv_backward_w_op.cc
rename to python/jittor/extern/mkl/ops/mkl_conv_backward_w_op.cc
diff --git a/extern/mkl/ops/mkl_conv_backward_w_op.h b/python/jittor/extern/mkl/ops/mkl_conv_backward_w_op.h
similarity index 100%
rename from extern/mkl/ops/mkl_conv_backward_w_op.h
rename to python/jittor/extern/mkl/ops/mkl_conv_backward_w_op.h
diff --git a/extern/mkl/ops/mkl_conv_backward_x_op.cc b/python/jittor/extern/mkl/ops/mkl_conv_backward_x_op.cc
similarity index 100%
rename from extern/mkl/ops/mkl_conv_backward_x_op.cc
rename to python/jittor/extern/mkl/ops/mkl_conv_backward_x_op.cc
diff --git a/extern/mkl/ops/mkl_conv_backward_x_op.h b/python/jittor/extern/mkl/ops/mkl_conv_backward_x_op.h
similarity index 100%
rename from extern/mkl/ops/mkl_conv_backward_x_op.h
rename to python/jittor/extern/mkl/ops/mkl_conv_backward_x_op.h
diff --git a/extern/mkl/ops/mkl_conv_op.cc b/python/jittor/extern/mkl/ops/mkl_conv_op.cc
similarity index 100%
rename from extern/mkl/ops/mkl_conv_op.cc
rename to python/jittor/extern/mkl/ops/mkl_conv_op.cc
diff --git a/extern/mkl/ops/mkl_conv_op.h b/python/jittor/extern/mkl/ops/mkl_conv_op.h
similarity index 100%
rename from extern/mkl/ops/mkl_conv_op.h
rename to python/jittor/extern/mkl/ops/mkl_conv_op.h
diff --git a/extern/mkl/ops/mkl_matmul_op.cc b/python/jittor/extern/mkl/ops/mkl_matmul_op.cc
similarity index 100%
rename from extern/mkl/ops/mkl_matmul_op.cc
rename to python/jittor/extern/mkl/ops/mkl_matmul_op.cc
diff --git a/extern/mkl/ops/mkl_matmul_op.h b/python/jittor/extern/mkl/ops/mkl_matmul_op.h
similarity index 100%
rename from extern/mkl/ops/mkl_matmul_op.h
rename to python/jittor/extern/mkl/ops/mkl_matmul_op.h
diff --git a/extern/mkl/ops/mkl_test_op.cc b/python/jittor/extern/mkl/ops/mkl_test_op.cc
similarity index 100%
rename from extern/mkl/ops/mkl_test_op.cc
rename to python/jittor/extern/mkl/ops/mkl_test_op.cc
diff --git a/extern/mkl/ops/mkl_test_op.h b/python/jittor/extern/mkl/ops/mkl_test_op.h
similarity index 100%
rename from extern/mkl/ops/mkl_test_op.h
rename to python/jittor/extern/mkl/ops/mkl_test_op.h
diff --git a/extern/mpi/inc/mpi_warper.h b/python/jittor/extern/mpi/inc/mpi_warper.h
similarity index 100%
rename from extern/mpi/inc/mpi_warper.h
rename to python/jittor/extern/mpi/inc/mpi_warper.h
diff --git a/extern/mpi/ops/mpi_all_reduce_op.cc b/python/jittor/extern/mpi/ops/mpi_all_reduce_op.cc
similarity index 100%
rename from extern/mpi/ops/mpi_all_reduce_op.cc
rename to python/jittor/extern/mpi/ops/mpi_all_reduce_op.cc
diff --git a/extern/mpi/ops/mpi_all_reduce_op.h b/python/jittor/extern/mpi/ops/mpi_all_reduce_op.h
similarity index 100%
rename from extern/mpi/ops/mpi_all_reduce_op.h
rename to python/jittor/extern/mpi/ops/mpi_all_reduce_op.h
diff --git a/extern/mpi/ops/mpi_broadcast_op.cc b/python/jittor/extern/mpi/ops/mpi_broadcast_op.cc
similarity index 100%
rename from extern/mpi/ops/mpi_broadcast_op.cc
rename to python/jittor/extern/mpi/ops/mpi_broadcast_op.cc
diff --git a/extern/mpi/ops/mpi_broadcast_op.h b/python/jittor/extern/mpi/ops/mpi_broadcast_op.h
similarity index 100%
rename from extern/mpi/ops/mpi_broadcast_op.h
rename to python/jittor/extern/mpi/ops/mpi_broadcast_op.h
diff --git a/extern/mpi/ops/mpi_reduce_op.cc b/python/jittor/extern/mpi/ops/mpi_reduce_op.cc
similarity index 100%
rename from extern/mpi/ops/mpi_reduce_op.cc
rename to python/jittor/extern/mpi/ops/mpi_reduce_op.cc
diff --git a/extern/mpi/ops/mpi_reduce_op.h b/python/jittor/extern/mpi/ops/mpi_reduce_op.h
similarity index 100%
rename from extern/mpi/ops/mpi_reduce_op.h
rename to python/jittor/extern/mpi/ops/mpi_reduce_op.h
diff --git a/extern/mpi/ops/mpi_test_op.cc b/python/jittor/extern/mpi/ops/mpi_test_op.cc
similarity index 100%
rename from extern/mpi/ops/mpi_test_op.cc
rename to python/jittor/extern/mpi/ops/mpi_test_op.cc
diff --git a/extern/mpi/ops/mpi_test_op.h b/python/jittor/extern/mpi/ops/mpi_test_op.h
similarity index 100%
rename from extern/mpi/ops/mpi_test_op.h
rename to python/jittor/extern/mpi/ops/mpi_test_op.h
diff --git a/extern/mpi/src/mpi_warper.cc b/python/jittor/extern/mpi/src/mpi_warper.cc
similarity index 100%
rename from extern/mpi/src/mpi_warper.cc
rename to python/jittor/extern/mpi/src/mpi_warper.cc
diff --git a/notebook/ConditionGAN.src.md b/python/jittor/notebook/ConditionGAN.src.md
similarity index 100%
rename from notebook/ConditionGAN.src.md
rename to python/jittor/notebook/ConditionGAN.src.md
diff --git a/notebook/LSGAN.src.md b/python/jittor/notebook/LSGAN.src.md
similarity index 100%
rename from notebook/LSGAN.src.md
rename to python/jittor/notebook/LSGAN.src.md
diff --git a/notebook/__main__.py b/python/jittor/notebook/__main__.py
similarity index 100%
rename from notebook/__main__.py
rename to python/jittor/notebook/__main__.py
diff --git a/notebook/basics.src.md b/python/jittor/notebook/basics.src.md
similarity index 100%
rename from notebook/basics.src.md
rename to python/jittor/notebook/basics.src.md
diff --git a/notebook/custom_op.src.md b/python/jittor/notebook/custom_op.src.md
similarity index 100%
rename from notebook/custom_op.src.md
rename to python/jittor/notebook/custom_op.src.md
diff --git a/notebook/example.src.md b/python/jittor/notebook/example.src.md
similarity index 100%
rename from notebook/example.src.md
rename to python/jittor/notebook/example.src.md
diff --git a/notebook/figs/mop.svg b/python/jittor/notebook/figs/mop.svg
similarity index 100%
rename from notebook/figs/mop.svg
rename to python/jittor/notebook/figs/mop.svg
diff --git a/notebook/md_to_ipynb.py b/python/jittor/notebook/md_to_ipynb.py
similarity index 100%
rename from notebook/md_to_ipynb.py
rename to python/jittor/notebook/md_to_ipynb.py
diff --git a/notebook/meta_op.src.md b/python/jittor/notebook/meta_op.src.md
similarity index 100%
rename from notebook/meta_op.src.md
rename to python/jittor/notebook/meta_op.src.md
diff --git a/notebook/profiler.src.md b/python/jittor/notebook/profiler.src.md
similarity index 100%
rename from notebook/profiler.src.md
rename to python/jittor/notebook/profiler.src.md
diff --git a/script/Dockerfile_cuda11 b/python/jittor/script/Dockerfile_cuda11
similarity index 100%
rename from script/Dockerfile_cuda11
rename to python/jittor/script/Dockerfile_cuda11
diff --git a/script/converter_server.sh b/python/jittor/script/converter_server.sh
similarity index 100%
rename from script/converter_server.sh
rename to python/jittor/script/converter_server.sh
diff --git a/script/inference_perf.py b/python/jittor/script/inference_perf.py
similarity index 100%
rename from script/inference_perf.py
rename to python/jittor/script/inference_perf.py
diff --git a/script/install.sh b/python/jittor/script/install.sh
similarity index 100%
rename from script/install.sh
rename to python/jittor/script/install.sh
diff --git a/script/install_llvm.sh b/python/jittor/script/install_llvm.sh
similarity index 100%
rename from script/install_llvm.sh
rename to python/jittor/script/install_llvm.sh
diff --git a/script/install_mkl.sh b/python/jittor/script/install_mkl.sh
similarity index 100%
rename from script/install_mkl.sh
rename to python/jittor/script/install_mkl.sh
diff --git a/script/make_doc.py b/python/jittor/script/make_doc.py
similarity index 100%
rename from script/make_doc.py
rename to python/jittor/script/make_doc.py
diff --git a/script/tmpi b/python/jittor/script/tmpi
similarity index 100%
rename from script/tmpi
rename to python/jittor/script/tmpi
diff --git a/script/update.sh b/python/jittor/script/update.sh
similarity index 100%
rename from script/update.sh
rename to python/jittor/script/update.sh
diff --git a/src/common.h b/python/jittor/src/common.h
similarity index 100%
rename from src/common.h
rename to python/jittor/src/common.h
diff --git a/src/core.h b/python/jittor/src/core.h
similarity index 100%
rename from src/core.h
rename to python/jittor/src/core.h
diff --git a/src/event_queue.cc b/python/jittor/src/event_queue.cc
similarity index 100%
rename from src/event_queue.cc
rename to python/jittor/src/event_queue.cc
diff --git a/src/event_queue.h b/python/jittor/src/event_queue.h
similarity index 100%
rename from src/event_queue.h
rename to python/jittor/src/event_queue.h
diff --git a/src/executor.cc b/python/jittor/src/executor.cc
similarity index 100%
rename from src/executor.cc
rename to python/jittor/src/executor.cc
diff --git a/src/executor.h b/python/jittor/src/executor.h
similarity index 100%
rename from src/executor.h
rename to python/jittor/src/executor.h
diff --git a/src/fused_op.cc b/python/jittor/src/fused_op.cc
similarity index 100%
rename from src/fused_op.cc
rename to python/jittor/src/fused_op.cc
diff --git a/src/fused_op.h b/python/jittor/src/fused_op.h
similarity index 100%
rename from src/fused_op.h
rename to python/jittor/src/fused_op.h
diff --git a/src/fuser.h b/python/jittor/src/fuser.h
similarity index 100%
rename from src/fuser.h
rename to python/jittor/src/fuser.h
diff --git a/src/grad.cc b/python/jittor/src/grad.cc
similarity index 100%
rename from src/grad.cc
rename to python/jittor/src/grad.cc
diff --git a/src/grad.h b/python/jittor/src/grad.h
similarity index 100%
rename from src/grad.h
rename to python/jittor/src/grad.h
diff --git a/src/graph.cc b/python/jittor/src/graph.cc
similarity index 100%
rename from src/graph.cc
rename to python/jittor/src/graph.cc
diff --git a/src/graph.h b/python/jittor/src/graph.h
similarity index 100%
rename from src/graph.h
rename to python/jittor/src/graph.h
diff --git a/src/init.cc b/python/jittor/src/init.cc
similarity index 100%
rename from src/init.cc
rename to python/jittor/src/init.cc
diff --git a/src/init.h b/python/jittor/src/init.h
similarity index 100%
rename from src/init.h
rename to python/jittor/src/init.h
diff --git a/src/jit_compiler.cc b/python/jittor/src/jit_compiler.cc
similarity index 100%
rename from src/jit_compiler.cc
rename to python/jittor/src/jit_compiler.cc
diff --git a/src/jit_compiler.h b/python/jittor/src/jit_compiler.h
similarity index 100%
rename from src/jit_compiler.h
rename to python/jittor/src/jit_compiler.h
diff --git a/src/jit_key.cc b/python/jittor/src/jit_key.cc
similarity index 100%
rename from src/jit_key.cc
rename to python/jittor/src/jit_key.cc
diff --git a/src/jit_key.h b/python/jittor/src/jit_key.h
similarity index 100%
rename from src/jit_key.h
rename to python/jittor/src/jit_key.h
diff --git a/src/lock.cc b/python/jittor/src/lock.cc
similarity index 100%
rename from src/lock.cc
rename to python/jittor/src/lock.cc
diff --git a/src/lock.h b/python/jittor/src/lock.h
similarity index 100%
rename from src/lock.h
rename to python/jittor/src/lock.h
diff --git a/src/mem/allocator.cc b/python/jittor/src/mem/allocator.cc
similarity index 100%
rename from src/mem/allocator.cc
rename to python/jittor/src/mem/allocator.cc
diff --git a/src/mem/allocator.h b/python/jittor/src/mem/allocator.h
similarity index 100%
rename from src/mem/allocator.h
rename to python/jittor/src/mem/allocator.h
diff --git a/src/mem/allocator/aligned_allocator.cc b/python/jittor/src/mem/allocator/aligned_allocator.cc
similarity index 100%
rename from src/mem/allocator/aligned_allocator.cc
rename to python/jittor/src/mem/allocator/aligned_allocator.cc
diff --git a/src/mem/allocator/aligned_allocator.h b/python/jittor/src/mem/allocator/aligned_allocator.h
similarity index 100%
rename from src/mem/allocator/aligned_allocator.h
rename to python/jittor/src/mem/allocator/aligned_allocator.h
diff --git a/src/mem/allocator/cuda_device_allocator.cc b/python/jittor/src/mem/allocator/cuda_device_allocator.cc
similarity index 100%
rename from src/mem/allocator/cuda_device_allocator.cc
rename to python/jittor/src/mem/allocator/cuda_device_allocator.cc
diff --git a/src/mem/allocator/cuda_device_allocator.h b/python/jittor/src/mem/allocator/cuda_device_allocator.h
similarity index 100%
rename from src/mem/allocator/cuda_device_allocator.h
rename to python/jittor/src/mem/allocator/cuda_device_allocator.h
diff --git a/src/mem/allocator/cuda_dual_allocator.cc b/python/jittor/src/mem/allocator/cuda_dual_allocator.cc
similarity index 100%
rename from src/mem/allocator/cuda_dual_allocator.cc
rename to python/jittor/src/mem/allocator/cuda_dual_allocator.cc
diff --git a/src/mem/allocator/cuda_dual_allocator.h b/python/jittor/src/mem/allocator/cuda_dual_allocator.h
similarity index 100%
rename from src/mem/allocator/cuda_dual_allocator.h
rename to python/jittor/src/mem/allocator/cuda_dual_allocator.h
diff --git a/src/mem/allocator/cuda_host_allocator.cc b/python/jittor/src/mem/allocator/cuda_host_allocator.cc
similarity index 100%
rename from src/mem/allocator/cuda_host_allocator.cc
rename to python/jittor/src/mem/allocator/cuda_host_allocator.cc
diff --git a/src/mem/allocator/cuda_host_allocator.h b/python/jittor/src/mem/allocator/cuda_host_allocator.h
similarity index 100%
rename from src/mem/allocator/cuda_host_allocator.h
rename to python/jittor/src/mem/allocator/cuda_host_allocator.h
diff --git a/src/mem/allocator/cuda_managed_allocator.cc b/python/jittor/src/mem/allocator/cuda_managed_allocator.cc
similarity index 100%
rename from src/mem/allocator/cuda_managed_allocator.cc
rename to python/jittor/src/mem/allocator/cuda_managed_allocator.cc
diff --git a/src/mem/allocator/cuda_managed_allocator.h b/python/jittor/src/mem/allocator/cuda_managed_allocator.h
similarity index 100%
rename from src/mem/allocator/cuda_managed_allocator.h
rename to python/jittor/src/mem/allocator/cuda_managed_allocator.h
diff --git a/src/mem/allocator/nfef_allocator.cc b/python/jittor/src/mem/allocator/nfef_allocator.cc
similarity index 100%
rename from src/mem/allocator/nfef_allocator.cc
rename to python/jittor/src/mem/allocator/nfef_allocator.cc
diff --git a/src/mem/allocator/nfef_allocator.h b/python/jittor/src/mem/allocator/nfef_allocator.h
similarity index 100%
rename from src/mem/allocator/nfef_allocator.h
rename to python/jittor/src/mem/allocator/nfef_allocator.h
diff --git a/src/mem/allocator/sfrl_allocator.cc b/python/jittor/src/mem/allocator/sfrl_allocator.cc
similarity index 100%
rename from src/mem/allocator/sfrl_allocator.cc
rename to python/jittor/src/mem/allocator/sfrl_allocator.cc
diff --git a/src/mem/allocator/sfrl_allocator.h b/python/jittor/src/mem/allocator/sfrl_allocator.h
similarity index 100%
rename from src/mem/allocator/sfrl_allocator.h
rename to python/jittor/src/mem/allocator/sfrl_allocator.h
diff --git a/src/mem/allocator/stat_allocator.cc b/python/jittor/src/mem/allocator/stat_allocator.cc
similarity index 100%
rename from src/mem/allocator/stat_allocator.cc
rename to python/jittor/src/mem/allocator/stat_allocator.cc
diff --git a/src/mem/allocator/stat_allocator.h b/python/jittor/src/mem/allocator/stat_allocator.h
similarity index 100%
rename from src/mem/allocator/stat_allocator.h
rename to python/jittor/src/mem/allocator/stat_allocator.h
diff --git a/src/mem/allocator/temp_allocator.cc b/python/jittor/src/mem/allocator/temp_allocator.cc
similarity index 100%
rename from src/mem/allocator/temp_allocator.cc
rename to python/jittor/src/mem/allocator/temp_allocator.cc
diff --git a/src/mem/allocator/temp_allocator.h b/python/jittor/src/mem/allocator/temp_allocator.h
similarity index 100%
rename from src/mem/allocator/temp_allocator.h
rename to python/jittor/src/mem/allocator/temp_allocator.h
diff --git a/src/mem/mem_info.cc b/python/jittor/src/mem/mem_info.cc
similarity index 100%
rename from src/mem/mem_info.cc
rename to python/jittor/src/mem/mem_info.cc
diff --git a/src/mem/mem_info.h b/python/jittor/src/mem/mem_info.h
similarity index 100%
rename from src/mem/mem_info.h
rename to python/jittor/src/mem/mem_info.h
diff --git a/src/memory_profiler.cc b/python/jittor/src/memory_profiler.cc
similarity index 100%
rename from src/memory_profiler.cc
rename to python/jittor/src/memory_profiler.cc
diff --git a/src/memory_profiler.h b/python/jittor/src/memory_profiler.h
similarity index 100%
rename from src/memory_profiler.h
rename to python/jittor/src/memory_profiler.h
diff --git a/src/misc/cpu_atomic.cc b/python/jittor/src/misc/cpu_atomic.cc
similarity index 100%
rename from src/misc/cpu_atomic.cc
rename to python/jittor/src/misc/cpu_atomic.cc
diff --git a/src/misc/cpu_atomic.h b/python/jittor/src/misc/cpu_atomic.h
similarity index 100%
rename from src/misc/cpu_atomic.h
rename to python/jittor/src/misc/cpu_atomic.h
diff --git a/src/misc/cstr.h b/python/jittor/src/misc/cstr.h
similarity index 100%
rename from src/misc/cstr.h
rename to python/jittor/src/misc/cstr.h
diff --git a/src/misc/cuda_atomic.h b/python/jittor/src/misc/cuda_atomic.h
similarity index 100%
rename from src/misc/cuda_atomic.h
rename to python/jittor/src/misc/cuda_atomic.h
diff --git a/src/misc/cuda_flags.cc b/python/jittor/src/misc/cuda_flags.cc
similarity index 100%
rename from src/misc/cuda_flags.cc
rename to python/jittor/src/misc/cuda_flags.cc
diff --git a/src/misc/cuda_flags.h b/python/jittor/src/misc/cuda_flags.h
similarity index 100%
rename from src/misc/cuda_flags.h
rename to python/jittor/src/misc/cuda_flags.h
diff --git a/src/misc/cuda_limits.h b/python/jittor/src/misc/cuda_limits.h
similarity index 100%
rename from src/misc/cuda_limits.h
rename to python/jittor/src/misc/cuda_limits.h
diff --git a/src/misc/deleter.h b/python/jittor/src/misc/deleter.h
similarity index 100%
rename from src/misc/deleter.h
rename to python/jittor/src/misc/deleter.h
diff --git a/src/misc/fast_shared_ptr.h b/python/jittor/src/misc/fast_shared_ptr.h
similarity index 100%
rename from src/misc/fast_shared_ptr.h
rename to python/jittor/src/misc/fast_shared_ptr.h
diff --git a/src/misc/hash.h b/python/jittor/src/misc/hash.h
similarity index 100%
rename from src/misc/hash.h
rename to python/jittor/src/misc/hash.h
diff --git a/src/misc/nan_checker.cc b/python/jittor/src/misc/nan_checker.cc
similarity index 100%
rename from src/misc/nan_checker.cc
rename to python/jittor/src/misc/nan_checker.cc
diff --git a/src/misc/nan_checker.cu b/python/jittor/src/misc/nan_checker.cu
similarity index 100%
rename from src/misc/nan_checker.cu
rename to python/jittor/src/misc/nan_checker.cu
diff --git a/src/misc/nan_checker.h b/python/jittor/src/misc/nan_checker.h
similarity index 100%
rename from src/misc/nan_checker.h
rename to python/jittor/src/misc/nan_checker.h
diff --git a/src/misc/nano_string.cc b/python/jittor/src/misc/nano_string.cc
similarity index 100%
rename from src/misc/nano_string.cc
rename to python/jittor/src/misc/nano_string.cc
diff --git a/src/misc/nano_string.h b/python/jittor/src/misc/nano_string.h
similarity index 100%
rename from src/misc/nano_string.h
rename to python/jittor/src/misc/nano_string.h
diff --git a/src/misc/nano_vector.h b/python/jittor/src/misc/nano_vector.h
similarity index 100%
rename from src/misc/nano_vector.h
rename to python/jittor/src/misc/nano_vector.h
diff --git a/src/misc/ring_buffer.cc b/python/jittor/src/misc/ring_buffer.cc
similarity index 100%
rename from src/misc/ring_buffer.cc
rename to python/jittor/src/misc/ring_buffer.cc
diff --git a/src/misc/ring_buffer.h b/python/jittor/src/misc/ring_buffer.h
similarity index 100%
rename from src/misc/ring_buffer.h
rename to python/jittor/src/misc/ring_buffer.h
diff --git a/src/misc/stack_vector.h b/python/jittor/src/misc/stack_vector.h
similarity index 100%
rename from src/misc/stack_vector.h
rename to python/jittor/src/misc/stack_vector.h
diff --git a/src/misc/str_utils.h b/python/jittor/src/misc/str_utils.h
similarity index 100%
rename from src/misc/str_utils.h
rename to python/jittor/src/misc/str_utils.h
diff --git a/src/misc/string_view_map.h b/python/jittor/src/misc/string_view_map.h
similarity index 100%
rename from src/misc/string_view_map.h
rename to python/jittor/src/misc/string_view_map.h
diff --git a/src/node.h b/python/jittor/src/node.h
similarity index 100%
rename from src/node.h
rename to python/jittor/src/node.h
diff --git a/src/numpy_func.h b/python/jittor/src/numpy_func.h
similarity index 100%
rename from src/numpy_func.h
rename to python/jittor/src/numpy_func.h
diff --git a/src/op.cc b/python/jittor/src/op.cc
similarity index 100%
rename from src/op.cc
rename to python/jittor/src/op.cc
diff --git a/src/op.h b/python/jittor/src/op.h
similarity index 100%
rename from src/op.h
rename to python/jittor/src/op.h
diff --git a/src/op_compiler.cc b/python/jittor/src/op_compiler.cc
similarity index 100%
rename from src/op_compiler.cc
rename to python/jittor/src/op_compiler.cc
diff --git a/src/op_compiler.h b/python/jittor/src/op_compiler.h
similarity index 100%
rename from src/op_compiler.h
rename to python/jittor/src/op_compiler.h
diff --git a/src/ops/arg_reduce_op.cc b/python/jittor/src/ops/arg_reduce_op.cc
similarity index 100%
rename from src/ops/arg_reduce_op.cc
rename to python/jittor/src/ops/arg_reduce_op.cc
diff --git a/src/ops/arg_reduce_op.h b/python/jittor/src/ops/arg_reduce_op.h
similarity index 100%
rename from src/ops/arg_reduce_op.h
rename to python/jittor/src/ops/arg_reduce_op.h
diff --git a/src/ops/argsort_op.cc b/python/jittor/src/ops/argsort_op.cc
similarity index 100%
rename from src/ops/argsort_op.cc
rename to python/jittor/src/ops/argsort_op.cc
diff --git a/src/ops/argsort_op.h b/python/jittor/src/ops/argsort_op.h
similarity index 100%
rename from src/ops/argsort_op.h
rename to python/jittor/src/ops/argsort_op.h
diff --git a/src/ops/array_op.cc b/python/jittor/src/ops/array_op.cc
similarity index 100%
rename from src/ops/array_op.cc
rename to python/jittor/src/ops/array_op.cc
diff --git a/src/ops/array_op.h b/python/jittor/src/ops/array_op.h
similarity index 100%
rename from src/ops/array_op.h
rename to python/jittor/src/ops/array_op.h
diff --git a/src/ops/binary_op.cc b/python/jittor/src/ops/binary_op.cc
similarity index 100%
rename from src/ops/binary_op.cc
rename to python/jittor/src/ops/binary_op.cc
diff --git a/src/ops/binary_op.h b/python/jittor/src/ops/binary_op.h
similarity index 100%
rename from src/ops/binary_op.h
rename to python/jittor/src/ops/binary_op.h
diff --git a/src/ops/binary_op_defs.h b/python/jittor/src/ops/binary_op_defs.h
similarity index 100%
rename from src/ops/binary_op_defs.h
rename to python/jittor/src/ops/binary_op_defs.h
diff --git a/src/ops/broadcast_to_op.cc b/python/jittor/src/ops/broadcast_to_op.cc
similarity index 100%
rename from src/ops/broadcast_to_op.cc
rename to python/jittor/src/ops/broadcast_to_op.cc
diff --git a/src/ops/broadcast_to_op.h b/python/jittor/src/ops/broadcast_to_op.h
similarity index 100%
rename from src/ops/broadcast_to_op.h
rename to python/jittor/src/ops/broadcast_to_op.h
diff --git a/src/ops/candidate_op.cc b/python/jittor/src/ops/candidate_op.cc
similarity index 100%
rename from src/ops/candidate_op.cc
rename to python/jittor/src/ops/candidate_op.cc
diff --git a/src/ops/candidate_op.h b/python/jittor/src/ops/candidate_op.h
similarity index 100%
rename from src/ops/candidate_op.h
rename to python/jittor/src/ops/candidate_op.h
diff --git a/src/ops/clone_op.cc b/python/jittor/src/ops/clone_op.cc
similarity index 100%
rename from src/ops/clone_op.cc
rename to python/jittor/src/ops/clone_op.cc
diff --git a/src/ops/clone_op.h b/python/jittor/src/ops/clone_op.h
similarity index 100%
rename from src/ops/clone_op.h
rename to python/jittor/src/ops/clone_op.h
diff --git a/src/ops/code_op.cc b/python/jittor/src/ops/code_op.cc
similarity index 100%
rename from src/ops/code_op.cc
rename to python/jittor/src/ops/code_op.cc
diff --git a/src/ops/code_op.h b/python/jittor/src/ops/code_op.h
similarity index 100%
rename from src/ops/code_op.h
rename to python/jittor/src/ops/code_op.h
diff --git a/src/ops/copy_op.cc b/python/jittor/src/ops/copy_op.cc
similarity index 100%
rename from src/ops/copy_op.cc
rename to python/jittor/src/ops/copy_op.cc
diff --git a/src/ops/copy_op.h b/python/jittor/src/ops/copy_op.h
similarity index 100%
rename from src/ops/copy_op.h
rename to python/jittor/src/ops/copy_op.h
diff --git a/src/ops/empty_op.cc b/python/jittor/src/ops/empty_op.cc
similarity index 100%
rename from src/ops/empty_op.cc
rename to python/jittor/src/ops/empty_op.cc
diff --git a/src/ops/empty_op.h b/python/jittor/src/ops/empty_op.h
similarity index 100%
rename from src/ops/empty_op.h
rename to python/jittor/src/ops/empty_op.h
diff --git a/src/ops/fetch_op.cc b/python/jittor/src/ops/fetch_op.cc
similarity index 100%
rename from src/ops/fetch_op.cc
rename to python/jittor/src/ops/fetch_op.cc
diff --git a/src/ops/fetch_op.h b/python/jittor/src/ops/fetch_op.h
similarity index 100%
rename from src/ops/fetch_op.h
rename to python/jittor/src/ops/fetch_op.h
diff --git a/src/ops/getitem_op.cc b/python/jittor/src/ops/getitem_op.cc
similarity index 100%
rename from src/ops/getitem_op.cc
rename to python/jittor/src/ops/getitem_op.cc
diff --git a/src/ops/getitem_op.h b/python/jittor/src/ops/getitem_op.h
similarity index 100%
rename from src/ops/getitem_op.h
rename to python/jittor/src/ops/getitem_op.h
diff --git a/src/ops/index_op.cc b/python/jittor/src/ops/index_op.cc
similarity index 100%
rename from src/ops/index_op.cc
rename to python/jittor/src/ops/index_op.cc
diff --git a/src/ops/index_op.h b/python/jittor/src/ops/index_op.h
similarity index 100%
rename from src/ops/index_op.h
rename to python/jittor/src/ops/index_op.h
diff --git a/src/ops/numpy_code_op.cc b/python/jittor/src/ops/numpy_code_op.cc
similarity index 100%
rename from src/ops/numpy_code_op.cc
rename to python/jittor/src/ops/numpy_code_op.cc
diff --git a/src/ops/numpy_code_op.h b/python/jittor/src/ops/numpy_code_op.h
similarity index 100%
rename from src/ops/numpy_code_op.h
rename to python/jittor/src/ops/numpy_code_op.h
diff --git a/src/ops/op_register.cc b/python/jittor/src/ops/op_register.cc
similarity index 100%
rename from src/ops/op_register.cc
rename to python/jittor/src/ops/op_register.cc
diff --git a/src/ops/op_register.h b/python/jittor/src/ops/op_register.h
similarity index 100%
rename from src/ops/op_register.h
rename to python/jittor/src/ops/op_register.h
diff --git a/src/ops/op_utils.cc b/python/jittor/src/ops/op_utils.cc
similarity index 100%
rename from src/ops/op_utils.cc
rename to python/jittor/src/ops/op_utils.cc
diff --git a/src/ops/random_op.cc b/python/jittor/src/ops/random_op.cc
similarity index 100%
rename from src/ops/random_op.cc
rename to python/jittor/src/ops/random_op.cc
diff --git a/src/ops/random_op.h b/python/jittor/src/ops/random_op.h
similarity index 100%
rename from src/ops/random_op.h
rename to python/jittor/src/ops/random_op.h
diff --git a/src/ops/reduce_op.cc b/python/jittor/src/ops/reduce_op.cc
similarity index 100%
rename from src/ops/reduce_op.cc
rename to python/jittor/src/ops/reduce_op.cc
diff --git a/src/ops/reduce_op.h b/python/jittor/src/ops/reduce_op.h
similarity index 100%
rename from src/ops/reduce_op.h
rename to python/jittor/src/ops/reduce_op.h
diff --git a/src/ops/reindex_op.cc b/python/jittor/src/ops/reindex_op.cc
similarity index 100%
rename from src/ops/reindex_op.cc
rename to python/jittor/src/ops/reindex_op.cc
diff --git a/src/ops/reindex_op.h b/python/jittor/src/ops/reindex_op.h
similarity index 100%
rename from src/ops/reindex_op.h
rename to python/jittor/src/ops/reindex_op.h
diff --git a/src/ops/reindex_reduce_op.cc b/python/jittor/src/ops/reindex_reduce_op.cc
similarity index 100%
rename from src/ops/reindex_reduce_op.cc
rename to python/jittor/src/ops/reindex_reduce_op.cc
diff --git a/src/ops/reindex_reduce_op.h b/python/jittor/src/ops/reindex_reduce_op.h
similarity index 100%
rename from src/ops/reindex_reduce_op.h
rename to python/jittor/src/ops/reindex_reduce_op.h
diff --git a/src/ops/reshape_op.cc b/python/jittor/src/ops/reshape_op.cc
similarity index 100%
rename from src/ops/reshape_op.cc
rename to python/jittor/src/ops/reshape_op.cc
diff --git a/src/ops/reshape_op.h b/python/jittor/src/ops/reshape_op.h
similarity index 100%
rename from src/ops/reshape_op.h
rename to python/jittor/src/ops/reshape_op.h
diff --git a/src/ops/setitem_op.cc b/python/jittor/src/ops/setitem_op.cc
similarity index 100%
rename from src/ops/setitem_op.cc
rename to python/jittor/src/ops/setitem_op.cc
diff --git a/src/ops/setitem_op.h b/python/jittor/src/ops/setitem_op.h
similarity index 100%
rename from src/ops/setitem_op.h
rename to python/jittor/src/ops/setitem_op.h
diff --git a/src/ops/tape_op.cc b/python/jittor/src/ops/tape_op.cc
similarity index 100%
rename from src/ops/tape_op.cc
rename to python/jittor/src/ops/tape_op.cc
diff --git a/src/ops/tape_op.h b/python/jittor/src/ops/tape_op.h
similarity index 100%
rename from src/ops/tape_op.h
rename to python/jittor/src/ops/tape_op.h
diff --git a/src/ops/ternary_op.cc b/python/jittor/src/ops/ternary_op.cc
similarity index 100%
rename from src/ops/ternary_op.cc
rename to python/jittor/src/ops/ternary_op.cc
diff --git a/src/ops/ternary_op.h b/python/jittor/src/ops/ternary_op.h
similarity index 100%
rename from src/ops/ternary_op.h
rename to python/jittor/src/ops/ternary_op.h
diff --git a/src/ops/transpose_op.cc b/python/jittor/src/ops/transpose_op.cc
similarity index 100%
rename from src/ops/transpose_op.cc
rename to python/jittor/src/ops/transpose_op.cc
diff --git a/src/ops/transpose_op.h b/python/jittor/src/ops/transpose_op.h
similarity index 100%
rename from src/ops/transpose_op.h
rename to python/jittor/src/ops/transpose_op.h
diff --git a/src/ops/unary_op.cc b/python/jittor/src/ops/unary_op.cc
similarity index 100%
rename from src/ops/unary_op.cc
rename to python/jittor/src/ops/unary_op.cc
diff --git a/src/ops/unary_op.h b/python/jittor/src/ops/unary_op.h
similarity index 100%
rename from src/ops/unary_op.h
rename to python/jittor/src/ops/unary_op.h
diff --git a/src/ops/unary_op_defs.h b/python/jittor/src/ops/unary_op_defs.h
similarity index 100%
rename from src/ops/unary_op_defs.h
rename to python/jittor/src/ops/unary_op_defs.h
diff --git a/src/ops/where_op.cc b/python/jittor/src/ops/where_op.cc
similarity index 100%
rename from src/ops/where_op.cc
rename to python/jittor/src/ops/where_op.cc
diff --git a/src/ops/where_op.h b/python/jittor/src/ops/where_op.h
similarity index 100%
rename from src/ops/where_op.h
rename to python/jittor/src/ops/where_op.h
diff --git a/src/opt/expr.cc b/python/jittor/src/opt/expr.cc
similarity index 100%
rename from src/opt/expr.cc
rename to python/jittor/src/opt/expr.cc
diff --git a/src/opt/expr.h b/python/jittor/src/opt/expr.h
similarity index 100%
rename from src/opt/expr.h
rename to python/jittor/src/opt/expr.h
diff --git a/src/opt/gopt/setitem_gopt.cc b/python/jittor/src/opt/gopt/setitem_gopt.cc
similarity index 100%
rename from src/opt/gopt/setitem_gopt.cc
rename to python/jittor/src/opt/gopt/setitem_gopt.cc
diff --git a/src/opt/jit_searcher.cc b/python/jittor/src/opt/jit_searcher.cc
similarity index 100%
rename from src/opt/jit_searcher.cc
rename to python/jittor/src/opt/jit_searcher.cc
diff --git a/src/opt/jit_searcher.h b/python/jittor/src/opt/jit_searcher.h
similarity index 100%
rename from src/opt/jit_searcher.h
rename to python/jittor/src/opt/jit_searcher.h
diff --git a/src/opt/kernel_ir.cc b/python/jittor/src/opt/kernel_ir.cc
similarity index 100%
rename from src/opt/kernel_ir.cc
rename to python/jittor/src/opt/kernel_ir.cc
diff --git a/src/opt/kernel_ir.h b/python/jittor/src/opt/kernel_ir.h
similarity index 100%
rename from src/opt/kernel_ir.h
rename to python/jittor/src/opt/kernel_ir.h
diff --git a/src/opt/pass/assume_aligned_pass.cc b/python/jittor/src/opt/pass/assume_aligned_pass.cc
similarity index 100%
rename from src/opt/pass/assume_aligned_pass.cc
rename to python/jittor/src/opt/pass/assume_aligned_pass.cc
diff --git a/src/opt/pass/assume_aligned_pass.h b/python/jittor/src/opt/pass/assume_aligned_pass.h
similarity index 100%
rename from src/opt/pass/assume_aligned_pass.h
rename to python/jittor/src/opt/pass/assume_aligned_pass.h
diff --git a/src/opt/pass/atomic_tuner_pass.h b/python/jittor/src/opt/pass/atomic_tuner_pass.h
similarity index 100%
rename from src/opt/pass/atomic_tuner_pass.h
rename to python/jittor/src/opt/pass/atomic_tuner_pass.h
diff --git a/src/opt/pass/check_cache_pass.cc b/python/jittor/src/opt/pass/check_cache_pass.cc
similarity index 100%
rename from src/opt/pass/check_cache_pass.cc
rename to python/jittor/src/opt/pass/check_cache_pass.cc
diff --git a/src/opt/pass/check_cache_pass.h b/python/jittor/src/opt/pass/check_cache_pass.h
similarity index 100%
rename from src/opt/pass/check_cache_pass.h
rename to python/jittor/src/opt/pass/check_cache_pass.h
diff --git a/src/opt/pass/compile_shapes_pass.cc b/python/jittor/src/opt/pass/compile_shapes_pass.cc
similarity index 100%
rename from src/opt/pass/compile_shapes_pass.cc
rename to python/jittor/src/opt/pass/compile_shapes_pass.cc
diff --git a/src/opt/pass/compile_shapes_pass.h b/python/jittor/src/opt/pass/compile_shapes_pass.h
similarity index 100%
rename from src/opt/pass/compile_shapes_pass.h
rename to python/jittor/src/opt/pass/compile_shapes_pass.h
diff --git a/src/opt/pass/const_var_pass.cc b/python/jittor/src/opt/pass/const_var_pass.cc
similarity index 100%
rename from src/opt/pass/const_var_pass.cc
rename to python/jittor/src/opt/pass/const_var_pass.cc
diff --git a/src/opt/pass/const_var_pass.h b/python/jittor/src/opt/pass/const_var_pass.h
similarity index 100%
rename from src/opt/pass/const_var_pass.h
rename to python/jittor/src/opt/pass/const_var_pass.h
diff --git a/src/opt/pass/expand_empty_block_pass.cc b/python/jittor/src/opt/pass/expand_empty_block_pass.cc
similarity index 100%
rename from src/opt/pass/expand_empty_block_pass.cc
rename to python/jittor/src/opt/pass/expand_empty_block_pass.cc
diff --git a/src/opt/pass/expand_empty_block_pass.h b/python/jittor/src/opt/pass/expand_empty_block_pass.h
similarity index 100%
rename from src/opt/pass/expand_empty_block_pass.h
rename to python/jittor/src/opt/pass/expand_empty_block_pass.h
diff --git a/src/opt/pass/fake_main_pass.cc b/python/jittor/src/opt/pass/fake_main_pass.cc
similarity index 100%
rename from src/opt/pass/fake_main_pass.cc
rename to python/jittor/src/opt/pass/fake_main_pass.cc
diff --git a/src/opt/pass/fake_main_pass.h b/python/jittor/src/opt/pass/fake_main_pass.h
similarity index 100%
rename from src/opt/pass/fake_main_pass.h
rename to python/jittor/src/opt/pass/fake_main_pass.h
diff --git a/src/opt/pass/float_atomic_fix_pass.cc b/python/jittor/src/opt/pass/float_atomic_fix_pass.cc
similarity index 100%
rename from src/opt/pass/float_atomic_fix_pass.cc
rename to python/jittor/src/opt/pass/float_atomic_fix_pass.cc
diff --git a/src/opt/pass/float_atomic_fix_pass.h b/python/jittor/src/opt/pass/float_atomic_fix_pass.h
similarity index 100%
rename from src/opt/pass/float_atomic_fix_pass.h
rename to python/jittor/src/opt/pass/float_atomic_fix_pass.h
diff --git a/src/opt/pass/insert_profile_loop_pass.cc b/python/jittor/src/opt/pass/insert_profile_loop_pass.cc
similarity index 100%
rename from src/opt/pass/insert_profile_loop_pass.cc
rename to python/jittor/src/opt/pass/insert_profile_loop_pass.cc
diff --git a/src/opt/pass/insert_profile_loop_pass.h b/python/jittor/src/opt/pass/insert_profile_loop_pass.h
similarity index 100%
rename from src/opt/pass/insert_profile_loop_pass.h
rename to python/jittor/src/opt/pass/insert_profile_loop_pass.h
diff --git a/src/opt/pass/loop_to_func_pass.cc b/python/jittor/src/opt/pass/loop_to_func_pass.cc
similarity index 100%
rename from src/opt/pass/loop_to_func_pass.cc
rename to python/jittor/src/opt/pass/loop_to_func_pass.cc
diff --git a/src/opt/pass/loop_to_func_pass.h b/python/jittor/src/opt/pass/loop_to_func_pass.h
similarity index 100%
rename from src/opt/pass/loop_to_func_pass.h
rename to python/jittor/src/opt/pass/loop_to_func_pass.h
diff --git a/src/opt/pass/loop_var_analyze_pass.cc b/python/jittor/src/opt/pass/loop_var_analyze_pass.cc
similarity index 100%
rename from src/opt/pass/loop_var_analyze_pass.cc
rename to python/jittor/src/opt/pass/loop_var_analyze_pass.cc
diff --git a/src/opt/pass/loop_var_analyze_pass.h b/python/jittor/src/opt/pass/loop_var_analyze_pass.h
similarity index 100%
rename from src/opt/pass/loop_var_analyze_pass.h
rename to python/jittor/src/opt/pass/loop_var_analyze_pass.h
diff --git a/src/opt/pass/mark_raw_pass.cc b/python/jittor/src/opt/pass/mark_raw_pass.cc
similarity index 100%
rename from src/opt/pass/mark_raw_pass.cc
rename to python/jittor/src/opt/pass/mark_raw_pass.cc
diff --git a/src/opt/pass/mark_raw_pass.h b/python/jittor/src/opt/pass/mark_raw_pass.h
similarity index 100%
rename from src/opt/pass/mark_raw_pass.h
rename to python/jittor/src/opt/pass/mark_raw_pass.h
diff --git a/src/opt/pass/merge_loop_pass.cc b/python/jittor/src/opt/pass/merge_loop_pass.cc
similarity index 100%
rename from src/opt/pass/merge_loop_pass.cc
rename to python/jittor/src/opt/pass/merge_loop_pass.cc
diff --git a/src/opt/pass/merge_loop_pass.h b/python/jittor/src/opt/pass/merge_loop_pass.h
similarity index 100%
rename from src/opt/pass/merge_loop_pass.h
rename to python/jittor/src/opt/pass/merge_loop_pass.h
diff --git a/src/opt/pass/merge_loop_var_pass.cc b/python/jittor/src/opt/pass/merge_loop_var_pass.cc
similarity index 100%
rename from src/opt/pass/merge_loop_var_pass.cc
rename to python/jittor/src/opt/pass/merge_loop_var_pass.cc
diff --git a/src/opt/pass/merge_loop_var_pass.h b/python/jittor/src/opt/pass/merge_loop_var_pass.h
similarity index 100%
rename from src/opt/pass/merge_loop_var_pass.h
rename to python/jittor/src/opt/pass/merge_loop_var_pass.h
diff --git a/src/opt/pass/parallel_pass.h b/python/jittor/src/opt/pass/parallel_pass.h
similarity index 100%
rename from src/opt/pass/parallel_pass.h
rename to python/jittor/src/opt/pass/parallel_pass.h
diff --git a/src/opt/pass/pass.cc b/python/jittor/src/opt/pass/pass.cc
similarity index 100%
rename from src/opt/pass/pass.cc
rename to python/jittor/src/opt/pass/pass.cc
diff --git a/src/opt/pass/pass.h b/python/jittor/src/opt/pass/pass.h
similarity index 100%
rename from src/opt/pass/pass.h
rename to python/jittor/src/opt/pass/pass.h
diff --git a/src/opt/pass/remove_intermediate_pass.cc b/python/jittor/src/opt/pass/remove_intermediate_pass.cc
similarity index 100%
rename from src/opt/pass/remove_intermediate_pass.cc
rename to python/jittor/src/opt/pass/remove_intermediate_pass.cc
diff --git a/src/opt/pass/remove_intermediate_pass.h b/python/jittor/src/opt/pass/remove_intermediate_pass.h
similarity index 100%
rename from src/opt/pass/remove_intermediate_pass.h
rename to python/jittor/src/opt/pass/remove_intermediate_pass.h
diff --git a/src/opt/pass/remove_loop_pass.cc b/python/jittor/src/opt/pass/remove_loop_pass.cc
similarity index 100%
rename from src/opt/pass/remove_loop_pass.cc
rename to python/jittor/src/opt/pass/remove_loop_pass.cc
diff --git a/src/opt/pass/remove_loop_pass.h b/python/jittor/src/opt/pass/remove_loop_pass.h
similarity index 100%
rename from src/opt/pass/remove_loop_pass.h
rename to python/jittor/src/opt/pass/remove_loop_pass.h
diff --git a/src/opt/pass/rename_loop_index_pass.cc b/python/jittor/src/opt/pass/rename_loop_index_pass.cc
similarity index 100%
rename from src/opt/pass/rename_loop_index_pass.cc
rename to python/jittor/src/opt/pass/rename_loop_index_pass.cc
diff --git a/src/opt/pass/rename_loop_index_pass.h b/python/jittor/src/opt/pass/rename_loop_index_pass.h
similarity index 100%
rename from src/opt/pass/rename_loop_index_pass.h
rename to python/jittor/src/opt/pass/rename_loop_index_pass.h
diff --git a/src/opt/pass/reorder_loop_pass.cc b/python/jittor/src/opt/pass/reorder_loop_pass.cc
similarity index 100%
rename from src/opt/pass/reorder_loop_pass.cc
rename to python/jittor/src/opt/pass/reorder_loop_pass.cc
diff --git a/src/opt/pass/reorder_loop_pass.h b/python/jittor/src/opt/pass/reorder_loop_pass.h
similarity index 100%
rename from src/opt/pass/reorder_loop_pass.h
rename to python/jittor/src/opt/pass/reorder_loop_pass.h
diff --git a/src/opt/pass/replace_for_num_pass.cc b/python/jittor/src/opt/pass/replace_for_num_pass.cc
similarity index 100%
rename from src/opt/pass/replace_for_num_pass.cc
rename to python/jittor/src/opt/pass/replace_for_num_pass.cc
diff --git a/src/opt/pass/replace_for_num_pass.h b/python/jittor/src/opt/pass/replace_for_num_pass.h
similarity index 100%
rename from src/opt/pass/replace_for_num_pass.h
rename to python/jittor/src/opt/pass/replace_for_num_pass.h
diff --git a/src/opt/pass/restride_pass.cc b/python/jittor/src/opt/pass/restride_pass.cc
similarity index 100%
rename from src/opt/pass/restride_pass.cc
rename to python/jittor/src/opt/pass/restride_pass.cc
diff --git a/src/opt/pass/restride_pass.h b/python/jittor/src/opt/pass/restride_pass.h
similarity index 100%
rename from src/opt/pass/restride_pass.h
rename to python/jittor/src/opt/pass/restride_pass.h
diff --git a/src/opt/pass/solve_conflict_define_pass.cc b/python/jittor/src/opt/pass/solve_conflict_define_pass.cc
similarity index 100%
rename from src/opt/pass/solve_conflict_define_pass.cc
rename to python/jittor/src/opt/pass/solve_conflict_define_pass.cc
diff --git a/src/opt/pass/solve_conflict_define_pass.h b/python/jittor/src/opt/pass/solve_conflict_define_pass.h
similarity index 100%
rename from src/opt/pass/solve_conflict_define_pass.h
rename to python/jittor/src/opt/pass/solve_conflict_define_pass.h
diff --git a/src/opt/pass/split_loop_pass.cc b/python/jittor/src/opt/pass/split_loop_pass.cc
similarity index 100%
rename from src/opt/pass/split_loop_pass.cc
rename to python/jittor/src/opt/pass/split_loop_pass.cc
diff --git a/src/opt/pass/split_loop_pass.h b/python/jittor/src/opt/pass/split_loop_pass.h
similarity index 100%
rename from src/opt/pass/split_loop_pass.h
rename to python/jittor/src/opt/pass/split_loop_pass.h
diff --git a/src/opt/pass/unroll_pass.cc b/python/jittor/src/opt/pass/unroll_pass.cc
similarity index 100%
rename from src/opt/pass/unroll_pass.cc
rename to python/jittor/src/opt/pass/unroll_pass.cc
diff --git a/src/opt/pass/unroll_pass.h b/python/jittor/src/opt/pass/unroll_pass.h
similarity index 100%
rename from src/opt/pass/unroll_pass.h
rename to python/jittor/src/opt/pass/unroll_pass.h
diff --git a/src/opt/pass/use_movnt_pass.cc b/python/jittor/src/opt/pass/use_movnt_pass.cc
similarity index 100%
rename from src/opt/pass/use_movnt_pass.cc
rename to python/jittor/src/opt/pass/use_movnt_pass.cc
diff --git a/src/opt/pass/use_movnt_pass.h b/python/jittor/src/opt/pass/use_movnt_pass.h
similarity index 100%
rename from src/opt/pass/use_movnt_pass.h
rename to python/jittor/src/opt/pass/use_movnt_pass.h
diff --git a/src/opt/pass/vectorize_pass.cc b/python/jittor/src/opt/pass/vectorize_pass.cc
similarity index 100%
rename from src/opt/pass/vectorize_pass.cc
rename to python/jittor/src/opt/pass/vectorize_pass.cc
diff --git a/src/opt/pass/vectorize_pass.h b/python/jittor/src/opt/pass/vectorize_pass.h
similarity index 100%
rename from src/opt/pass/vectorize_pass.h
rename to python/jittor/src/opt/pass/vectorize_pass.h
diff --git a/src/opt/pass_manager.cc b/python/jittor/src/opt/pass_manager.cc
similarity index 100%
rename from src/opt/pass_manager.cc
rename to python/jittor/src/opt/pass_manager.cc
diff --git a/src/opt/pass_manager.h b/python/jittor/src/opt/pass_manager.h
similarity index 100%
rename from src/opt/pass_manager.h
rename to python/jittor/src/opt/pass_manager.h
diff --git a/src/opt/tuner/broadcast_tuner.cc b/python/jittor/src/opt/tuner/broadcast_tuner.cc
similarity index 100%
rename from src/opt/tuner/broadcast_tuner.cc
rename to python/jittor/src/opt/tuner/broadcast_tuner.cc
diff --git a/src/opt/tuner/broadcast_tuner.h b/python/jittor/src/opt/tuner/broadcast_tuner.h
similarity index 100%
rename from src/opt/tuner/broadcast_tuner.h
rename to python/jittor/src/opt/tuner/broadcast_tuner.h
diff --git a/src/opt/tuner/conv_tuner.cc b/python/jittor/src/opt/tuner/conv_tuner.cc
similarity index 100%
rename from src/opt/tuner/conv_tuner.cc
rename to python/jittor/src/opt/tuner/conv_tuner.cc
diff --git a/src/opt/tuner/conv_tuner.h b/python/jittor/src/opt/tuner/conv_tuner.h
similarity index 100%
rename from src/opt/tuner/conv_tuner.h
rename to python/jittor/src/opt/tuner/conv_tuner.h
diff --git a/src/opt/tuner/matmul_tuner.cc b/python/jittor/src/opt/tuner/matmul_tuner.cc
similarity index 100%
rename from src/opt/tuner/matmul_tuner.cc
rename to python/jittor/src/opt/tuner/matmul_tuner.cc
diff --git a/src/opt/tuner/matmul_tuner.h b/python/jittor/src/opt/tuner/matmul_tuner.h
similarity index 100%
rename from src/opt/tuner/matmul_tuner.h
rename to python/jittor/src/opt/tuner/matmul_tuner.h
diff --git a/src/opt/tuner/reduce_tuner.cc b/python/jittor/src/opt/tuner/reduce_tuner.cc
similarity index 100%
rename from src/opt/tuner/reduce_tuner.cc
rename to python/jittor/src/opt/tuner/reduce_tuner.cc
diff --git a/src/opt/tuner/reduce_tuner.h b/python/jittor/src/opt/tuner/reduce_tuner.h
similarity index 100%
rename from src/opt/tuner/reduce_tuner.h
rename to python/jittor/src/opt/tuner/reduce_tuner.h
diff --git a/src/opt/tuner/reorder_tuner.cc b/python/jittor/src/opt/tuner/reorder_tuner.cc
similarity index 100%
rename from src/opt/tuner/reorder_tuner.cc
rename to python/jittor/src/opt/tuner/reorder_tuner.cc
diff --git a/src/opt/tuner/reorder_tuner.h b/python/jittor/src/opt/tuner/reorder_tuner.h
similarity index 100%
rename from src/opt/tuner/reorder_tuner.h
rename to python/jittor/src/opt/tuner/reorder_tuner.h
diff --git a/src/opt/tuner/tuner.cc b/python/jittor/src/opt/tuner/tuner.cc
similarity index 100%
rename from src/opt/tuner/tuner.cc
rename to python/jittor/src/opt/tuner/tuner.cc
diff --git a/src/opt/tuner/tuner.h b/python/jittor/src/opt/tuner/tuner.h
similarity index 100%
rename from src/opt/tuner/tuner.h
rename to python/jittor/src/opt/tuner/tuner.h
diff --git a/src/opt/tuner_manager.cc b/python/jittor/src/opt/tuner_manager.cc
similarity index 100%
rename from src/opt/tuner_manager.cc
rename to python/jittor/src/opt/tuner_manager.cc
diff --git a/src/opt/tuner_manager.h b/python/jittor/src/opt/tuner_manager.h
similarity index 100%
rename from src/opt/tuner_manager.h
rename to python/jittor/src/opt/tuner_manager.h
diff --git a/src/opt/var_relay.cc b/python/jittor/src/opt/var_relay.cc
similarity index 100%
rename from src/opt/var_relay.cc
rename to python/jittor/src/opt/var_relay.cc
diff --git a/src/opt/var_relay.h b/python/jittor/src/opt/var_relay.h
similarity index 100%
rename from src/opt/var_relay.h
rename to python/jittor/src/opt/var_relay.h
diff --git a/src/parallel_compiler.cc b/python/jittor/src/parallel_compiler.cc
similarity index 100%
rename from src/parallel_compiler.cc
rename to python/jittor/src/parallel_compiler.cc
diff --git a/src/parallel_compiler.h b/python/jittor/src/parallel_compiler.h
similarity index 100%
rename from src/parallel_compiler.h
rename to python/jittor/src/parallel_compiler.h
diff --git a/src/profiler/cache_info.cc b/python/jittor/src/profiler/cache_info.cc
similarity index 100%
rename from src/profiler/cache_info.cc
rename to python/jittor/src/profiler/cache_info.cc
diff --git a/src/profiler/cache_info.h b/python/jittor/src/profiler/cache_info.h
similarity index 100%
rename from src/profiler/cache_info.h
rename to python/jittor/src/profiler/cache_info.h
diff --git a/src/profiler/memory_checker.cc b/python/jittor/src/profiler/memory_checker.cc
similarity index 100%
rename from src/profiler/memory_checker.cc
rename to python/jittor/src/profiler/memory_checker.cc
diff --git a/src/profiler/memory_checker.h b/python/jittor/src/profiler/memory_checker.h
similarity index 100%
rename from src/profiler/memory_checker.h
rename to python/jittor/src/profiler/memory_checker.h
diff --git a/src/profiler/profiler.cc b/python/jittor/src/profiler/profiler.cc
similarity index 100%
rename from src/profiler/profiler.cc
rename to python/jittor/src/profiler/profiler.cc
diff --git a/src/profiler/profiler.h b/python/jittor/src/profiler/profiler.h
similarity index 100%
rename from src/profiler/profiler.h
rename to python/jittor/src/profiler/profiler.h
diff --git a/src/profiler/profiler_guard.h b/python/jittor/src/profiler/profiler_guard.h
similarity index 100%
rename from src/profiler/profiler_guard.h
rename to python/jittor/src/profiler/profiler_guard.h
diff --git a/src/profiler/replacement.cc b/python/jittor/src/profiler/replacement.cc
similarity index 100%
rename from src/profiler/replacement.cc
rename to python/jittor/src/profiler/replacement.cc
diff --git a/src/profiler/replacement.h b/python/jittor/src/profiler/replacement.h
similarity index 100%
rename from src/profiler/replacement.h
rename to python/jittor/src/profiler/replacement.h
diff --git a/src/profiler/simple_profiler.h b/python/jittor/src/profiler/simple_profiler.h
similarity index 100%
rename from src/profiler/simple_profiler.h
rename to python/jittor/src/profiler/simple_profiler.h
diff --git a/src/profiler/vtop.cc b/python/jittor/src/profiler/vtop.cc
similarity index 100%
rename from src/profiler/vtop.cc
rename to python/jittor/src/profiler/vtop.cc
diff --git a/src/pybind/core.cc b/python/jittor/src/pybind/core.cc
similarity index 100%
rename from src/pybind/core.cc
rename to python/jittor/src/pybind/core.cc
diff --git a/src/pybind/py_var_tracer.cc b/python/jittor/src/pybind/py_var_tracer.cc
similarity index 100%
rename from src/pybind/py_var_tracer.cc
rename to python/jittor/src/pybind/py_var_tracer.cc
diff --git a/src/pybind/py_var_tracer.h b/python/jittor/src/pybind/py_var_tracer.h
similarity index 100%
rename from src/pybind/py_var_tracer.h
rename to python/jittor/src/pybind/py_var_tracer.h
diff --git a/src/pybind/py_var_tracer_interface.h b/python/jittor/src/pybind/py_var_tracer_interface.h
similarity index 100%
rename from src/pybind/py_var_tracer_interface.h
rename to python/jittor/src/pybind/py_var_tracer_interface.h
diff --git a/src/pyjt/numpy.cc b/python/jittor/src/pyjt/numpy.cc
similarity index 100%
rename from src/pyjt/numpy.cc
rename to python/jittor/src/pyjt/numpy.cc
diff --git a/src/pyjt/numpy.h b/python/jittor/src/pyjt/numpy.h
similarity index 100%
rename from src/pyjt/numpy.h
rename to python/jittor/src/pyjt/numpy.h
diff --git a/src/pyjt/py_arg_printer.cc b/python/jittor/src/pyjt/py_arg_printer.cc
similarity index 100%
rename from src/pyjt/py_arg_printer.cc
rename to python/jittor/src/pyjt/py_arg_printer.cc
diff --git a/src/pyjt/py_arg_printer.h b/python/jittor/src/pyjt/py_arg_printer.h
similarity index 100%
rename from src/pyjt/py_arg_printer.h
rename to python/jittor/src/pyjt/py_arg_printer.h
diff --git a/src/pyjt/py_array_op.cc b/python/jittor/src/pyjt/py_array_op.cc
similarity index 100%
rename from src/pyjt/py_array_op.cc
rename to python/jittor/src/pyjt/py_array_op.cc
diff --git a/src/pyjt/py_caller.cc b/python/jittor/src/pyjt/py_caller.cc
similarity index 100%
rename from src/pyjt/py_caller.cc
rename to python/jittor/src/pyjt/py_caller.cc
diff --git a/src/pyjt/py_caller.h b/python/jittor/src/pyjt/py_caller.h
similarity index 100%
rename from src/pyjt/py_caller.h
rename to python/jittor/src/pyjt/py_caller.h
diff --git a/src/pyjt/py_converter.h b/python/jittor/src/pyjt/py_converter.h
similarity index 100%
rename from src/pyjt/py_converter.h
rename to python/jittor/src/pyjt/py_converter.h
diff --git a/src/pyjt/py_obj_holder.h b/python/jittor/src/pyjt/py_obj_holder.h
similarity index 100%
rename from src/pyjt/py_obj_holder.h
rename to python/jittor/src/pyjt/py_obj_holder.h
diff --git a/src/pyjt/py_ring_buffer.cc b/python/jittor/src/pyjt/py_ring_buffer.cc
similarity index 100%
rename from src/pyjt/py_ring_buffer.cc
rename to python/jittor/src/pyjt/py_ring_buffer.cc
diff --git a/src/pyjt/py_ring_buffer.h b/python/jittor/src/pyjt/py_ring_buffer.h
similarity index 100%
rename from src/pyjt/py_ring_buffer.h
rename to python/jittor/src/pyjt/py_ring_buffer.h
diff --git a/src/pyjt/pyjt_console.h b/python/jittor/src/pyjt/pyjt_console.h
similarity index 100%
rename from src/pyjt/pyjt_console.h
rename to python/jittor/src/pyjt/pyjt_console.h
diff --git a/src/test/test_expr.cc b/python/jittor/src/test/test_expr.cc
similarity index 100%
rename from src/test/test_expr.cc
rename to python/jittor/src/test/test_expr.cc
diff --git a/src/test/test_fast_shared_ptr.cc b/python/jittor/src/test/test_fast_shared_ptr.cc
similarity index 100%
rename from src/test/test_fast_shared_ptr.cc
rename to python/jittor/src/test/test_fast_shared_ptr.cc
diff --git a/src/test/test_jit_key.cc b/python/jittor/src/test/test_jit_key.cc
similarity index 100%
rename from src/test/test_jit_key.cc
rename to python/jittor/src/test/test_jit_key.cc
diff --git a/src/test/test_kernel_ir.cc b/python/jittor/src/test/test_kernel_ir.cc
similarity index 100%
rename from src/test/test_kernel_ir.cc
rename to python/jittor/src/test/test_kernel_ir.cc
diff --git a/src/test/test_nano_vector.cc b/python/jittor/src/test/test_nano_vector.cc
similarity index 100%
rename from src/test/test_nano_vector.cc
rename to python/jittor/src/test/test_nano_vector.cc
diff --git a/src/test/test_op_compiler.cc b/python/jittor/src/test/test_op_compiler.cc
similarity index 100%
rename from src/test/test_op_compiler.cc
rename to python/jittor/src/test/test_op_compiler.cc
diff --git a/src/test/test_op_relay.cc b/python/jittor/src/test/test_op_relay.cc
similarity index 100%
rename from src/test/test_op_relay.cc
rename to python/jittor/src/test/test_op_relay.cc
diff --git a/src/test/test_setitem_op.cc b/python/jittor/src/test/test_setitem_op.cc
similarity index 100%
rename from src/test/test_setitem_op.cc
rename to python/jittor/src/test/test_setitem_op.cc
diff --git a/src/test/test_sfrl_allocator.cc b/python/jittor/src/test/test_sfrl_allocator.cc
similarity index 100%
rename from src/test/test_sfrl_allocator.cc
rename to python/jittor/src/test/test_sfrl_allocator.cc
diff --git a/src/types.h b/python/jittor/src/types.h
similarity index 100%
rename from src/types.h
rename to python/jittor/src/types.h
diff --git a/src/update_queue.cc b/python/jittor/src/update_queue.cc
similarity index 100%
rename from src/update_queue.cc
rename to python/jittor/src/update_queue.cc
diff --git a/src/update_queue.h b/python/jittor/src/update_queue.h
similarity index 100%
rename from src/update_queue.h
rename to python/jittor/src/update_queue.h
diff --git a/src/utils/cache_compile.cc b/python/jittor/src/utils/cache_compile.cc
similarity index 100%
rename from src/utils/cache_compile.cc
rename to python/jittor/src/utils/cache_compile.cc
diff --git a/src/utils/cache_compile.h b/python/jittor/src/utils/cache_compile.h
similarity index 100%
rename from src/utils/cache_compile.h
rename to python/jittor/src/utils/cache_compile.h
diff --git a/src/utils/flags.cc b/python/jittor/src/utils/flags.cc
similarity index 100%
rename from src/utils/flags.cc
rename to python/jittor/src/utils/flags.cc
diff --git a/src/utils/flags.h b/python/jittor/src/utils/flags.h
similarity index 100%
rename from src/utils/flags.h
rename to python/jittor/src/utils/flags.h
diff --git a/src/utils/jit_utils.cc b/python/jittor/src/utils/jit_utils.cc
similarity index 100%
rename from src/utils/jit_utils.cc
rename to python/jittor/src/utils/jit_utils.cc
diff --git a/src/utils/log.cc b/python/jittor/src/utils/log.cc
similarity index 100%
rename from src/utils/log.cc
rename to python/jittor/src/utils/log.cc
diff --git a/src/utils/log.h b/python/jittor/src/utils/log.h
similarity index 100%
rename from src/utils/log.h
rename to python/jittor/src/utils/log.h
diff --git a/src/utils/mwsr_list.cc b/python/jittor/src/utils/mwsr_list.cc
similarity index 100%
rename from src/utils/mwsr_list.cc
rename to python/jittor/src/utils/mwsr_list.cc
diff --git a/src/utils/mwsr_list.h b/python/jittor/src/utils/mwsr_list.h
similarity index 100%
rename from src/utils/mwsr_list.h
rename to python/jittor/src/utils/mwsr_list.h
diff --git a/src/utils/tracer.cc b/python/jittor/src/utils/tracer.cc
similarity index 100%
rename from src/utils/tracer.cc
rename to python/jittor/src/utils/tracer.cc
diff --git a/src/utils/tracer.h b/python/jittor/src/utils/tracer.h
similarity index 100%
rename from src/utils/tracer.h
rename to python/jittor/src/utils/tracer.h
diff --git a/src/var.cc b/python/jittor/src/var.cc
similarity index 100%
rename from src/var.cc
rename to python/jittor/src/var.cc
diff --git a/src/var.h b/python/jittor/src/var.h
similarity index 100%
rename from src/var.h
rename to python/jittor/src/var.h
diff --git a/src/var_holder.cc b/python/jittor/src/var_holder.cc
similarity index 100%
rename from src/var_holder.cc
rename to python/jittor/src/var_holder.cc
diff --git a/src/var_holder.h b/python/jittor/src/var_holder.h
similarity index 100%
rename from src/var_holder.h
rename to python/jittor/src/var_holder.h
diff --git a/src/var_slices.cc b/python/jittor/src/var_slices.cc
similarity index 100%
rename from src/var_slices.cc
rename to python/jittor/src/var_slices.cc
diff --git a/src/var_slices.h b/python/jittor/src/var_slices.h
similarity index 100%
rename from src/var_slices.h
rename to python/jittor/src/var_slices.h