Commit Graph

47 Commits

Author SHA1 Message Date
looop5 1a3da1fc47 update submodule akg 2022-11-03 19:53:47 +08:00
Zichun Ye b96d172b31 fix code check problem for custom op 2022-10-25 23:55:44 +08:00
Zichun Ye 26a70116f2 rename the docarator ms_kernel to kernel
fix code check problem

fix code check problem

fix code review comment
2022-10-20 16:05:12 +08:00
Zichun Ye df01cfcf93 update custom_aot_cpu/gpu_kernel_mod to new kernel mod class 2022-10-03 16:07:45 +08:00
zengzitao 3b87ff1bb2 lu ascend adapt 2022-09-29 14:08:33 +08:00
i-robot ecc476a9ff
!41114 Custom Op AOT attr support update
Merge pull request !41114 from zichun_ye/custom_attr_aot_update
2022-08-30 10:35:14 +00:00
Zichun Ye 567c6f48de update custom op aot attr support
fix code check problem
2022-08-30 00:42:12 +08:00
Zichun Ye bc63bdf6f2 update solve tri expander
update code via code review comments

update code via code review comments
2022-08-27 22:33:06 +08:00
zengzitao b4a7a8dad8 solve_triangular adapt 2022-08-18 14:15:46 +08:00
Zichun Ye 9b4a19b503 update mac test case 2022-07-04 16:01:22 +08:00
Zichun Ye 4279fd9413 update custom op aot mode with attr support
fix code check problem

fix code check problem

fix code check problem

disable compiling aot kernel on windows

update the custom aot op attr class

fix pclint problem

fix typo

update attr class based on code review comments

fix code check problem
2022-06-27 22:55:36 +08:00
i-robot b476009eba
!35312 Custom Op Hybrid and AOT mode update
Merge pull request !35312 from zichun_ye/fused_kernel_transformer
2022-06-17 22:11:09 +00:00
Zichun Ye 90cb42933a update aot auto-compile; add fused kernel to transformer
fix code checck problem

update aot compiling logic

udpate new intrin st cases

fix code check

fix typo
2022-06-15 16:43:52 +08:00
Zichun Ye 8de24b4d2f update ms_kernel related files
fix import problem

update code check

update test case

update test case

rename file
2022-06-10 15:41:38 +08:00
r1chardf1d0 15c7ba2087 julia only support x86_64 2022-05-31 10:15:13 +08:00
Yang Jiao d6afbbc732 fix custom bug on ascend 2022-05-24 09:41:00 +08:00
Zichun Ye cd20125db6 update aot custom op
update aot  custom op

fix code check problems
2022-05-13 12:08:17 +08:00
Zichun Ye 2d29ec8781 drop ms_hybrid cpu fp16 2022-05-10 10:12:34 +08:00
Zichun Ye d5322e6f5f add ms_hybrid cpu fp16 support
format file

update test case

fix test case on d

update test cases on D

update akg commit id

update test cases
2022-05-03 22:14:23 +08:00
Zichun Ye 9ebe2befe7 add custom hybrid dsl cpu support
fix typo

fix code check problem

update code check problem

add missing python module

update cpu test cases

update return result
2022-04-22 15:53:27 +08:00
ttudu adb18b82ca dynamic shape CPU & GPU 2022-03-22 13:03:31 +08:00
Zichun Ye 61469ad999 update ms hybrid test cases
fix ms hybrid doc

revert changes
2022-03-10 20:31:56 +08:00
Zichun Ye d8087a35e3 add ms_hybrid decorator; add new Custom mode ms_hybrid
fix code check error

update test cases, update akg version

drop grid for now

update testcase

update benchmark in test case

update test case name

update test cases

update test case

add input num parameterattr in json
2022-02-27 23:38:54 +08:00
r1chardf1d0 ce80c46edc add julia cache 2022-02-22 16:02:36 +08:00
i-robot e9b4b4ae61
!29555 Custom Op Inplace Assign
Merge pull request !29555 from jiaoy1224/customop
2022-02-17 11:21:27 +00:00
Yang Jiao b35cedeb62 hybrid inplace assign 2022-02-16 16:25:16 +08:00
r1chardf1d0 aef9e0841c julia test only support linux now 2022-02-16 15:08:40 +08:00
r1chardf1d0 6074fa8aae add julia error log 2022-02-15 09:50:16 +08:00
i-robot aae56dbae4
!28962 Change aot error code strategy and modify logs
Merge pull request !28962 from jiaoy1224/aot
2022-01-30 03:03:03 +00:00
Yang Jiao 7d4599e65a change aot func return code 2022-01-29 19:25:42 +08:00
r1chardf1d0 ed4624c879 add version check 2022-01-29 12:16:13 +08:00
r1chardf1d0 044e110f7a custom op support julia 2022-01-24 12:39:02 +08:00
i-robot 0138b6594a !28140 Fix aot test case
Merge pull request !28140 from jiaoy1224/aot
2021-12-24 10:02:03 +00:00
Yang Jiao d3824a6a6c fix aot test case 2021-12-24 15:31:11 +08:00
xulei 612cfe453b add c++ std to custom aot compile cmd 2021-12-20 14:18:50 +08:00
i-robot ed63220a9f !27357 Fix Type Error of Aot Test File on Mac and Add Comments
Merge pull request !27357 from jiaoy1224/arithmetic
2021-12-10 06:21:16 +00:00
Yang Jiao d9531a4f73 add head for aot test 2021-12-08 17:05:13 +08:00
looop5 ae5532fb9e pass lambda function to out_shape and out_dtype 2021-12-06 10:21:16 +08:00
Yang Jiao 5acce31c72 apply autodiff in custom 2021-12-03 11:16:19 +08:00
looop5 9c4fe8b6bf update akg, refine document of Custom op 2021-11-16 11:06:40 +08:00
looop5 58e27d87bc add Custom, custom_op_info_register, CustomRegOp to __init__ 2021-11-11 20:49:16 +08:00
Yang Jiao 2b112ef656 add custom pynative test cases 2021-11-08 17:37:39 +08:00
looop5 b89d744e80 Custom op supports no reg info 2021-11-05 17:21:39 +08:00
Yang Jiao 9bcf28d15d adapt custom op to pyfunc kernel 2021-10-31 22:08:10 +08:00
looop5 0b34bf33bc add custom op pynative testcases 2021-10-25 20:53:42 +08:00
Yang Jiao 40b648b873 add aot 2021-10-22 11:37:39 +08:00
looop5 46789f260a Custom operator supports tbe dsl 2021-10-15 16:28:33 +08:00