Commit Graph

5 Commits

Author SHA1 Message Date
Rattus
ede3d4b966 MPDynamic: Add support for model defined dtype
If the model defines a dtype that is different to what is in the state
dict, respect that at load time. This is done as part of the casting
process.
2026-01-27 18:57:40 +10:00
Rattus
01ca403bed ruff 2026-01-27 18:57:21 +10:00
Rattus
7a18963a33 misc cleanup 2026-01-27 18:57:19 +10:00
Rattus
e8c9977973 add missing del on unpin 2026-01-27 18:56:53 +10:00
Rattus
3c2ce0d58d pinned_memory: add python
Add a python for managing pinned memory of the weight/bias module level.
This allocates, pins and attached a tensor to a module for the pin for this
module. It does not set the weight, just allocates a singular ram buffer
for population and bulk DMA transfer.
2026-01-27 18:56:10 +10:00