lzhengning
/
JittorMirror
mirror of https://github.com/Jittor/Jittor
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

add updated code

This commit is contained in:
lidongyang 2025-07-15 20:05:00 +08:00
parent 2f37158e3e
commit b79ac22b05
2 changed files with 5 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
python/jittor/compile_extern.py
View File

@ -456,7 +456,8 @@ def setup_cutt():
def install_cutlass(root_folder):
# Modified from: https://github.com/ap-hynninen/cutlass
url = "https://cloud.tsinghua.edu.cn/f/171e49e5825549548bc4/?dl=1"
# url = "https://cloud.tsinghua.edu.cn/f/171e49e5825549548bc4/?dl=1"
url = "https://cg.cs.tsinghua.edu.cn/jittor/assets/cutlass.zip"
filename = "cutlass.zip"
fullname = os.path.join(root_folder, filename)

4
python/jittor/src/ops/getitem_op.cc
View File

@ -572,10 +572,12 @@ void GetitemOp::jit_run() {
index_t(vp@d[0 @for(j,0,VD,@if((VS@d>>j)&1, + i@{j+FOV} * vs@d@@s@j,))])
, ??? ))))));
)
@for(d, 0, IDIM, if (iid@d < 0) iid@d += ishape@d;
)
auto iid = 0 @for(d, 0, IDIM, + iid@d * istride@d);
op[oid] = ip[iid];
}
}
#endif // JIT
} // jittor
} // jittor