diff --git a/prebuild/aarch64/build_cce.cc.o b/prebuild/aarch64/build_cce.cc.o index a682f0499cf2a1266db404cb256f184023c3f7f8..a3b31f9e6ec8503a071952902b35a738cc42b9ef 100644 --- a/prebuild/aarch64/build_cce.cc.o +++ b/prebuild/aarch64/build_cce.cc.o @@ -1,3 +1,3 @@ version https://git-lfs.github.com/spec/v1 -oid sha256:00d543ed1d4446a9e1fa395471d61fad1c957e4d2d20675a60231a8619c52627 -size 808760 +oid sha256:35d68a57e6adb1ad04a7f7f29a9b3cb18cc56a0fd65303b3249c000a5f681ee3 +size 809168 diff --git a/prebuild/aarch64/cceconf.cc.o b/prebuild/aarch64/cceconf.cc.o index f2b46b897a1f32eb949dbc8429774800cae70de2..e08fe4ff104eb27413d6b04ff87d3dbee21958e5 100644 --- a/prebuild/aarch64/cceconf.cc.o +++ b/prebuild/aarch64/cceconf.cc.o @@ -1,3 +1,3 @@ version https://git-lfs.github.com/spec/v1 -oid sha256:6fda630a52f544debfc6134e9aea5b3acaa4a04db0075cc6f0dffa33ce32916c -size 165688 +oid sha256:8aba29022b5e5022edac4487892ee6b04517c74f9cd3a868e34b42d1caf73ed9 +size 168408 diff --git a/prebuild/aarch64/lib_info.txt b/prebuild/aarch64/lib_info.txt index cdc9a02413dc8ca37c60b6ebbd0b15af75bdcf24..a3da0c803099172b7af0f961355b30cc97810829 100644 --- a/prebuild/aarch64/lib_info.txt +++ b/prebuild/aarch64/lib_info.txt @@ -1,3 +1,3 @@ [lib information] git branch: master -commit id: 2b9614946e4a87bdec784dd907496bdfdcd80312 +commit id: 024fbfb801cd6bfdcfa4fe4203185f4aa2dcd17e diff --git a/prebuild/aarch64/paged_attention_tiling.cc.o b/prebuild/aarch64/paged_attention_tiling.cc.o index 2c57d0ed19395bba89d5be47733f5ad5527f000b..bc2d26df16f695c9ae613dcf8ffdd5627bcf9429 100644 --- a/prebuild/aarch64/paged_attention_tiling.cc.o +++ b/prebuild/aarch64/paged_attention_tiling.cc.o @@ -1,3 +1,3 @@ version https://git-lfs.github.com/spec/v1 -oid sha256:2bba3266077c8d3f28edd6006b5ad7f5d2a925f472eb6a1c56cdba8b7bb8d86c -size 119800 +oid sha256:ef2d4b83d7d06068535ddf4be6bc80a4e343db6baf9c93c9b2de4a9c3adcbd48 +size 125352 diff --git a/prebuild/aarch64/pp_matmul_add_tiling.cc.o b/prebuild/aarch64/pp_matmul_add_tiling.cc.o new file mode 100644 index 0000000000000000000000000000000000000000..684ab63c79725e5edf70c246b1da13a59501dbda Binary files /dev/null and b/prebuild/aarch64/pp_matmul_add_tiling.cc.o differ diff --git a/prebuild/aarch64/pp_matmul_common_tiling.cc.o b/prebuild/aarch64/pp_matmul_common_tiling.cc.o new file mode 100644 index 0000000000000000000000000000000000000000..a07a6707b458c842496569c880a611ccfbcef2a1 Binary files /dev/null and b/prebuild/aarch64/pp_matmul_common_tiling.cc.o differ diff --git a/prebuild/aarch64/pp_matmul_tiling.cc.o b/prebuild/aarch64/pp_matmul_tiling.cc.o index 622b34bfe9d9708f08f50de5d0f870192bd55c6d..70041c1c3282fd3bd99b6a18c3117daa1f43f0bc 100644 --- a/prebuild/aarch64/pp_matmul_tiling.cc.o +++ b/prebuild/aarch64/pp_matmul_tiling.cc.o @@ -1,3 +1,3 @@ version https://git-lfs.github.com/spec/v1 -oid sha256:c80c6aa1d10c4940b5b2bc03327f5c0268059a26837849e0b68af497a9ff03b0 -size 122256 +oid sha256:2bc3532457a227cbd15fdd38ad987f4757ae27cdca3b8e2dcc23f9969ae2157b +size 142064 diff --git a/prebuild/aarch64/update_cce_tiling.cc.o b/prebuild/aarch64/update_cce_tiling.cc.o index 18d530e56605b6562398172b1a575edf65743774..8241d71f34bab004c2f88844aaaaf0c76c8c5c2a 100644 --- a/prebuild/aarch64/update_cce_tiling.cc.o +++ b/prebuild/aarch64/update_cce_tiling.cc.o @@ -1,3 +1,3 @@ version https://git-lfs.github.com/spec/v1 -oid sha256:b42ca59b6e9125b0372ea93b3c6c61e43f3cd6e5f1d966a761ee1d8698f651d8 -size 377384 +oid sha256:1674f126e31416e067b095c7affe0393c1c9e6c0990642af8dbcf0e164a2b4db +size 386192 diff --git a/prebuild/x86_64/build_cce.cc.o b/prebuild/x86_64/build_cce.cc.o index 2275ea616e3a21bdf128e10a9e67ffe2cd1c58f7..9a687373a1d452313b1c38bc8daabb7f4c99cd1d 100644 --- a/prebuild/x86_64/build_cce.cc.o +++ b/prebuild/x86_64/build_cce.cc.o @@ -1,3 +1,3 @@ version https://git-lfs.github.com/spec/v1 -oid sha256:c54b18e538a0b1adcc286c689fe3142bcda18af12d2d7dd5900fb58906b1e37c -size 767784 +oid sha256:1a99aac73e6b63675caab040bc64a7881fc96abbdd3f523ae29d07bfb23ebc00 +size 768096 diff --git a/prebuild/x86_64/cceconf.cc.o b/prebuild/x86_64/cceconf.cc.o index bb31d5fe8a5093584c38c91261c2c622363a4cee..4f375bab78b9aa3796c26a447f63671d05aad042 100644 --- a/prebuild/x86_64/cceconf.cc.o +++ b/prebuild/x86_64/cceconf.cc.o @@ -1,3 +1,3 @@ version https://git-lfs.github.com/spec/v1 -oid sha256:cc8f782877577b5fae34cbb91798a263366777b714ec924abf13148e2a9fd097 -size 154880 +oid sha256:7501454501a5b920960291bbaccfc67c9b1e86fa25078f2371ba4a8d05341f4e +size 156864 diff --git a/prebuild/x86_64/lib_info.txt b/prebuild/x86_64/lib_info.txt index cdc9a02413dc8ca37c60b6ebbd0b15af75bdcf24..a3da0c803099172b7af0f961355b30cc97810829 100644 --- a/prebuild/x86_64/lib_info.txt +++ b/prebuild/x86_64/lib_info.txt @@ -1,3 +1,3 @@ [lib information] git branch: master -commit id: 2b9614946e4a87bdec784dd907496bdfdcd80312 +commit id: 024fbfb801cd6bfdcfa4fe4203185f4aa2dcd17e diff --git a/prebuild/x86_64/paged_attention_tiling.cc.o b/prebuild/x86_64/paged_attention_tiling.cc.o index 881fdea3a93fc4f9634f8d39b4b24d2d7a924113..a9694edc9df024d952cfa291e65ac4151b4f87c3 100644 --- a/prebuild/x86_64/paged_attention_tiling.cc.o +++ b/prebuild/x86_64/paged_attention_tiling.cc.o @@ -1,3 +1,3 @@ version https://git-lfs.github.com/spec/v1 -oid sha256:4ad042c8f38e52d9793ce71f500d466580d7abdcf9cf199b2cf1443247c50d0c -size 113584 +oid sha256:1c86565100823f9c0d17fff897b771e9742ee1976bfb68cfd715ce54969efc83 +size 117352 diff --git a/prebuild/x86_64/pp_matmul_add_tiling.cc.o b/prebuild/x86_64/pp_matmul_add_tiling.cc.o new file mode 100644 index 0000000000000000000000000000000000000000..aec004bf5404b97c4616d1ebd16b1052720abb27 Binary files /dev/null and b/prebuild/x86_64/pp_matmul_add_tiling.cc.o differ diff --git a/prebuild/x86_64/pp_matmul_common_tiling.cc.o b/prebuild/x86_64/pp_matmul_common_tiling.cc.o new file mode 100644 index 0000000000000000000000000000000000000000..4e1f7c1f102b917421bb2830783a2a6270ed15c4 Binary files /dev/null and b/prebuild/x86_64/pp_matmul_common_tiling.cc.o differ diff --git a/prebuild/x86_64/pp_matmul_tiling.cc.o b/prebuild/x86_64/pp_matmul_tiling.cc.o index 37f3612d3e5c9154479b33cc1802cfea9a147265..a7985defbb2d37a56939acbc36b0a0db6bab7068 100644 --- a/prebuild/x86_64/pp_matmul_tiling.cc.o +++ b/prebuild/x86_64/pp_matmul_tiling.cc.o @@ -1,3 +1,3 @@ version https://git-lfs.github.com/spec/v1 -oid sha256:a00d9ad67869e07048c9656d3a6ce49caf5c3e3043cc9e19746c0a58cfe06707 -size 116776 +oid sha256:eeb9a9be64c16f6f44a7bb0ebb598e901c6a7ce4eb5789d82ea49c8eaba95b14 +size 133400 diff --git a/prebuild/x86_64/update_cce_tiling.cc.o b/prebuild/x86_64/update_cce_tiling.cc.o index 704cd881d14ccf2e1b8e64e845cb9e53907752b4..5d3fa0bf1a089b9f9aaab7818f762c424212ed19 100644 --- a/prebuild/x86_64/update_cce_tiling.cc.o +++ b/prebuild/x86_64/update_cce_tiling.cc.o @@ -1,3 +1,3 @@ version https://git-lfs.github.com/spec/v1 -oid sha256:a1069232e6af7d1fe2cdc55eeff689e0d92d4466b84e6a11358406a293b403f1 -size 367168 +oid sha256:65d0a9d1e26a5c2d456653aee497dc8484f64a1d7df0e73d0dcd6ebae753aec2 +size 376136 diff --git a/python/akg/composite/build_module.py b/python/akg/composite/build_module.py index bb8111d310ca6644683a12af52af244f17e6a790..e20a9b518e34d034e8b87b2a8874984039f9a473 100644 --- a/python/akg/composite/build_module.py +++ b/python/akg/composite/build_module.py @@ -655,6 +655,10 @@ def _build_to_module_ascend(desc_s_in, desc_d_in, attr, use_repo=True): attr["pragma_modshift"] = True segment_tree, segment_infos = get_construct_args(desc_s_in, attr, post_funcs) + if desc_d_in.get("backend"): + backend_func = akg.tvm.get_global_func("cce.set_backend") + backend_func(desc_d_in["backend"]) + if desc_d_in.get("enable_cce_lib"): attr["enable_cce_lib"] = True return _build_to_module_ascend_lib(desc_s_in, kernel_name)