Skip to content

[IR][Script] Support indexing/slicing of both shared/global tensors#37

Merged
yaoyaoding merged 7 commits intomainfrom
yaoyao/tensor-slice-indexing
Sep 7, 2025
Merged

[IR][Script] Support indexing/slicing of both shared/global tensors#37
yaoyaoding merged 7 commits intomainfrom
yaoyao/tensor-slice-indexing

Conversation

@yaoyaoding
Copy link
Copy Markdown
Member

This PR adds the support of indexing and slicing of global and shared tensors.

New instructions added:

  • GlobalSlice
  • GlobalIndex
  • SharedSlice
  • SharedIndex

In Tilus Script, we can use the following syntax to indexing/slicing a shared/global tensor:

a: fp16[32, 64]
a[1, 2] # indexing, got fp16
a[1, :] # slice, got fp16[64]
a[:, 3] # slice, got fp16[32]

@copy-pr-bot
Copy link
Copy Markdown

copy-pr-bot Bot commented Sep 6, 2025

This pull request requires additional validation before any workflows can run on NVIDIA's runners.

Pull request vetters can view their responsibilities here.

Contributors can view more details about this message here.

@yaoyaoding yaoyaoding force-pushed the yaoyao/tensor-slice-indexing branch 2 times, most recently from 5ae0e3e to fb952ba Compare September 6, 2025 19:10
Signed-off-by: Yaoyao Ding <dingyaoyao.cs@gmail.com>
Signed-off-by: Yaoyao Ding <dingyaoyao.cs@gmail.com>
Signed-off-by: Yaoyao Ding <dingyaoyao.cs@gmail.com>
Signed-off-by: Yaoyao Ding <dingyaoyao.cs@gmail.com>
Signed-off-by: Yaoyao Ding <dingyaoyao.cs@gmail.com>
Signed-off-by: Yaoyao Ding <dingyaoyao.cs@gmail.com>
Signed-off-by: Yaoyao Ding <dingyaoyao.cs@gmail.com>
@yaoyaoding yaoyaoding force-pushed the yaoyao/tensor-slice-indexing branch from 8b28472 to 7627b59 Compare September 6, 2025 23:26
@yaoyaoding yaoyaoding merged commit 2579e7d into main Sep 7, 2025
8 checks passed
@yaoyaoding yaoyaoding deleted the yaoyao/tensor-slice-indexing branch September 10, 2025 20:25
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant