lzhengning
/
jittoromega
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity
15 Commits 2 Branches 0 Tags 249 KiB
C++ 95.2%
Python 2.3%
CMake 1.5%
C 0.6%
Shell 0.3%
Other 0.1%
Go to file
lzhengning 3cceb3a1a6 update a lot 2023-06-07 23:42:33 +08:00
include/omega update a lot 2023-06-07 23:42:33 +08:00
jittoromega partial commit 2023-05-26 14:51:03 +08:00
playground partial commit 2023-05-26 14:51:03 +08:00
src update a lot 2023-06-07 23:42:33 +08:00
test update a lot 2023-06-07 23:42:33 +08:00
.gitignore update a lot 2023-06-07 23:42:33 +08:00
CMakeLists.txt update a lot 2023-06-07 23:42:33 +08:00
README.md memset after alloc 2023-05-17 19:46:35 +08:00
omega_trainer.py update a lot 2023-06-03 15:03:40 +08:00
run_test.sh update a lot 2023-06-07 23:42:33 +08:00
test.py update a lot 2023-06-07 23:42:33 +08:00

README.md

Interface

  • preprare [sync, once]:

    • IN: List[int layer_id]
    • OUT: List[gpu0_buffer_ptr, gpu1_buffer_ptr, ..., gpu7_buffer_ptr]

  • ready [sync]: until ok

    • IN: int layer_id
    • OUT: return ok
    • ERROR: return others

  • release [async]

    • IN: layer id
    • OUT: return ok
    • ERROR: return others

  • grad_prepare [sync, once]

    • IN:
    • OUT:

  • grad_ready [sync]: return ok when the buffer_ptr is avaliable

    • IN: layer_id
    • OUT: return ok

  • grad_release [async]

    • IN: layer_id
    • OUT: return ok