Skip to content

feat(pto): use vdiv in ptoas - #16

Merged
KurrinQu merged 1 commit into
PTO-ISA:pto-devfrom
wangchen965:vdiv
Oct 26, 2026
Merged

KurrinQu merged 1 commit into
PTO-ISA:pto-devfrom
wangchen965:vdiv

Conversation

@wangchen965

@wangchen965 wangchen965 commented Oct 24, 2026 •

Copy link
Copy Markdown

summary

Previously, when tilelang was adapted for vmi, vmi did not support the input of precision. Therefore, the high-precision vdiv was implemented in a soft manner on the tilelang side.Now directly forward the "precision" parameter to Ptoas

Function and performance testing

The test was conducted using the engram_gate_fwd_vmi operator from TileKernels. The function was correct and the performance was comparable to that of asc.

tokens hidden save ASC_us VMI_us ratio
512 128 0 7.83 8.00 0.979
8001 128 0 101.15 100.98 1.002
512 7168 0 35.41 35.75 0.991
8001 7168 0 504.54 506.08 0.997
512 7168 1 45.58 45.84 0.994
8001 7168 1 528.89 527.33 1.003

Signed-off-by: wangchen <wangchen965@huawei.com>
@KurrinQu
KurrinQu merged commit 0acfde1 into PTO-ISA:pto-dev Oct 26, 2026
1 check passed
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.

2 participants