[WIP]feat(tensor): add vocab-parallel embedding_lookup operator-ops-tilelang-AtomGit

Star257

Star257

Draft

[WIP]feat(tensor): add vocab-parallel embedding_lookup operator #64

yizigeAdd on 2 days agoMerged fromWill_09/ops-tilelang： embedding_lookup into cann/ops-tilelang： master

[WIP]feat(tensor): add vocab-parallel embedding_lookup operator #64

Draft

yizigeAdd on 2 days ago

Merged fromWill_09/ops-tilelang embedding_lookupintocann/ops-tilelang master

Total: 5 files changed+807-0

Msrc/cann_ops_tilelang/__init__.py+6-0

Msrc/cann_ops_tilelang/tensor/__init__.py+6-0

Asrc/cann_ops_tilelang/tensor/embedding_lookup.py+171-0

Asrc/cann_ops_tilelang/tensor/embedding_lookup_kernel.py+252-0

Atests/tensor/test_embedding_lookup.py+372-0

File content is under review, please try again later