zhuzilin / pytorch-malloc
An external memory allocator example for PyTorch.
☆13Updated 2 years ago
Related projects: ⓘ
- Official resporitory for "IPDPS' 24 QSync: Quantization-Minimized Synchronous Distributed Training Across Hybrid Devices".☆19Updated 6 months ago
- Tacker: Tensor-CUDA Core Kernel Fusion for Improving the GPU Utilization while Ensuring QoS☆17Updated 2 years ago
- ☆11Updated 3 years ago
- An Attention Superoptimizer☆19Updated 4 months ago
- ☆14Updated last week
- Benchmark for matrix multiplications between dense and block sparse (BSR) matrix in TVM, blocksparse (Gray et al.) and cuSparse.☆24Updated 4 years ago
- Chameleon: Adaptive Code Optimization for Expedited Deep Neural Network Compilation☆26Updated 4 years ago
- PyTorch compilation tutorial covering TorchScript, torch.fx, and Slapo☆18Updated last year
- Benchmark PyTorch Custom Operators☆13Updated last year
- The code for our paper "Neural Architecture Search as Program Transformation Exploration"☆18Updated 3 years ago
- ☆7Updated last year
- ☆52Updated this week
- You Only Search Once: On Lightweight Differentiable Architecture Search for Resource-Constrained Embedded Platforms☆10Updated last year
- Artifact for IPDPS'21: DSXplore: Optimizing Convolutional Neural Networks via Sliding-Channel Convolutions.☆13Updated 3 years ago
- ☆18Updated 2 years ago
- Yet another Polyhedra Compiler for DeepLearning☆19Updated last year
- ☆19Updated 5 months ago
- GPTQ inference TVM kernel☆35Updated 4 months ago
- ☆17Updated 3 years ago
- Benchmark scripts for TVM☆73Updated 2 years ago
- (NeurIPS 2022) Automatically finding good model-parallel strategies, especially for complex models and clusters.☆33Updated last year
- ☆17Updated 3 years ago
- PSTensor provides a way to hack the memory management of tensors in TensorFlow and PyTorch by defining your own C++ Tensor Class.☆9Updated 2 years ago
- This is the implementation for paper: AdaTune: Adaptive Tensor Program CompilationMade Efficient (NeurIPS 2020).☆13Updated 3 years ago
- This is a repo which contains some details about how to use OpenCL backend (Xilinx/Intel).☆24Updated 4 years ago
- DietCode Code Release☆59Updated 2 years ago
- Slides from 2021-12-15 talk, "TVM Developer Bootcamp – Writing Hardware Backends"☆10Updated 2 years ago
- ☆20Updated last year
- A standalone GEMM kernel for fp16 activation and quantized weight, extracted from FasterTransformer☆82Updated 6 months ago