Skip to content

Tadd and softmax dynamic examples#3

Open
MirkoDeVita98 wants to merge 31 commits into
Zhendong404:pto-dsl-implfrom
MirkoDeVita98:based-on-pr-388
Open

Tadd and softmax dynamic examples#3
MirkoDeVita98 wants to merge 31 commits into
Zhendong404:pto-dsl-implfrom
MirkoDeVita98:based-on-pr-388

Conversation

@MirkoDeVita98
Copy link
Copy Markdown

No description provided.

Comment thread ptodsl/examples/jit/tadd_launch.py Outdated
Comment on lines 120 to 122
A: pto.tensor_spec(rank=2, dtype=pto.f32),
B: pto.tensor_spec(rank=2, dtype=pto.f32),
C: pto.tensor_spec(rank=2, dtype=pto.f32),
Copy link
Copy Markdown

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Does setting rank=2 or not affect correctness?

Copy link
Copy Markdown
Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

yes it works also without

@MirkoDeVita98 MirkoDeVita98 changed the title Tadd dynamic example Tadd and softmax dynamic examples May 26, 2026
scores: pto.tensor_spec(dtype=pto.f32),
out: pto.tensor_spec(dtype=pto.f32),
rows: pto.i32,
seq: pto.i32,
Copy link
Copy Markdown

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Better test a hybrid dimension here: dynamic dim for row and static closure for seq

@Zhendong404 Zhendong404 force-pushed the pto-dsl-impl branch 5 times, most recently from 83a9f18 to cc7eb51 Compare May 28, 2026 07:47
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.

4 participants