Commit Graph

14 Commits

Author SHA1 Message Date
wls2002
5cbe3c14bb Current Progress: After final design presentation 2023-06-19 15:17:56 +08:00
wls2002
acedd67617 change project structure and using .ini as config file 2023-06-15 11:05:26 +08:00
wls2002
bd421de9ad Current Progress: After final design presentation 2023-06-14 10:20:55 +08:00
wls2002
72c9d4167a FAST! 2023-05-13 20:58:03 +08:00
wls2002
6006f92f3f finish jit-able speciate function
next time i'll create a new branch
2023-05-12 19:26:02 +08:00
wls2002
9b56f4ff73 做了一些时间测试:
1. jit中的vmap一个函数不会触发重新编译
2. jax.lax.while_lop单独执行确实可以提前中断,但在vmap中的性能需要考虑。
2023-05-12 16:42:57 +08:00
wls2002
47b1a1dbb2 refactor genome.py use (C, 4) to replace (2, N, N) to represent connections
faster, faster and faster!
2023-05-12 00:57:55 +08:00
wls2002
e5fc1167d9 refactor genome.py use (C, 4) to replace (2, N, N) to represent connections
use "cons" to replace "connections" in code for beauty
2023-05-11 19:49:19 +08:00
wls2002
e2a5117554 initial commit in branch developing 2023-05-11 19:04:56 +08:00
wls2002
dde338696f reduce the use of device_get in speciate 2023-05-08 15:58:09 +08:00
wls2002
d4a75b9394 use jit().lower.compile in create functions 2023-05-08 02:35:04 +08:00
wls2002
497d89fc69 move o2o_distance and o2m_distance to pipelines 2023-05-08 01:19:45 +08:00
wls2002
73ac1bcfe0 generally complete, but not work well. Debug 2023-05-06 11:35:44 +08:00
wls2002
6faa07f507 initial commit 2023-05-05 14:19:13 +08:00