- Use Python environment with PyTorch and Flash2 or Flash3 installed
- Install helper matmul dispatching and transmission scheduler libraries:
make - Run the test training script:
python train.py
-
Notifications
You must be signed in to change notification settings - Fork 0
als244/awsm_dataflow
Folders and files
| Name | Name | Last commit message | Last commit date | |
|---|---|---|---|---|
Repository files navigation
About
Automatic Working Set Management for High Throughput ML Applications with Small Memory Footprint
Resources
Stars
Watchers
Forks
Releases
No releases published
Packages 0
No packages published