cmake -DLOGGING=ON .. && make
LD_PRELOAD=./libtma.so python
>>> import torch
>>> torch.rand(1,3)
[tma-trace]: torch-alloc-cpu required=[12] return=[16] bytes ptr=0x5e1b1f28d2c0
-
Notifications
You must be signed in to change notification settings - Fork 0
3cityrnd/libtma
Folders and files
Name | Name | Last commit message | Last commit date | |
---|---|---|---|---|
Repository files navigation
About
Lib Torch Memory Align
Resources
Stars
Watchers
Forks
Releases
No releases published
Packages 0
No packages published