feat(multimodal): add torch.compile support for five image ops-ops-tilelang-AtomGit

Star257

Star257

Open

feat(multimodal): add torch.compile support for five image ops #68

贝比诺Add on 1 day agoMerged frombabynull/ops-tilelang： codex/multimodal-torch-compile into cann/ops-tilelang： master

feat(multimodal): add torch.compile support for five image ops #68

Open

贝比诺Add on 1 day ago

Merged frombabynull/ops-tilelang codex/multimodal-torch-compileintocann/ops-tilelang master

Total: 5 files changed+489-40

Msrc/cann_ops_tilelang/multimodal/dsocr.py+23-3

Msrc/cann_ops_tilelang/multimodal/dsocr_kernel.py+1-1

Msrc/cann_ops_tilelang/multimodal/image_embed_ops.py+184-36

Mtests/multimodal/test_dsocr.py+37-0

Mtests/multimodal/test_image_embed_ops.py+244-0

File content is under review, please try again later