zhuzilin / pytorch-malloc
An external memory allocator example for PyTorch.
☆14Updated 3 years ago
Alternatives and similar repositories for pytorch-malloc:
Users that are interested in pytorch-malloc are comparing it to the libraries listed below
- Official resporitory for "IPDPS' 24 QSync: Quantization-Minimized Synchronous Distributed Training Across Hybrid Devices".☆19Updated last year
- An Attention Superoptimizer☆21Updated 3 months ago
- ☆11Updated 4 years ago
- Tacker: Tensor-CUDA Core Kernel Fusion for Improving the GPU Utilization while Ensuring QoS☆25Updated 2 months ago
- Benchmark PyTorch Custom Operators☆14Updated last year