WebIt addresses the kernel fusion problem of dynamic shapes with shape propagation and constraints collecting methods. This is the first work to demonstrate how to build an end-to-end dynamic shape compiler based on MLIR infrastructure. Experiments show that DISC achieves up to 3.3x speedup than TensorFlow/PyTorch, and 1.8x than Nimble. WebDISC: A dynamic shape compiler for machine learning workloads. K Zhu, WY Zhao, Z Zheng, TY Guo, PZ Zhao, JJ Bai, J Yang, XY Liu, ... Proceedings of the 1st Workshop on Machine Learning and Systems, 89-95, 2024. 8: 2024: Parameter-Efficient Sparsity for Large Language Models Fine-Tuning.
DISC: A Dynamic Shape Compiler for Machine Learning …
WebDISC: A Dynamic Shape Compiler for Machine Learning Workloads. Kai Zhu, Wenyi Zhao, Zhen Zheng, Tianyou Guo, Pengzhan Zhao, Junjie Bai, Jun Yang, Xiaoyou Liu, Lansong … WebMay 24, 2024 · Basically, I want to compile my DNN model (in PyTorch, ONNX, etc) with dynamic batch support. In other words, I want my compiled TVM module to process inputs with various batch sizes. For instance, I want my ResNet model to process inputs with sizes of [1, 3, 224, 224], [2, 3, 224, 224], and so on. I’ve seen many similar topics, but no one ... pct germany
DISC: A Dynamic Shape Compiler for Machine Learning Workloads
WebDOI: 10.1145/3437984.3458838 Corpus ID: 232168739; DISC: A Dynamic Shape Compiler for Machine Learning Workloads @article{Zhu2024DISCAD, title={DISC: A Dynamic Shape Compiler for Machine Learning Workloads}, author={Kai Zhu and Wenyi Zhao and Zhen Zheng and Tianyou Guo and Pengzhan Zhao and Junjie Bai and Jun … WebBy default, ONNX defines models in terms of dynamic shapes. The ONNX importer retains that dynamism upon import, and the compiler attempts to convert the model into a static shapes at compile time. If this fails, there may still be dynamic operations in the model. Not all TVM kernels currently support dynamic shapes, please file an issue on ... Webmization of dynamic shape ops. DISC shows a way to build a complete optimization system that targets dynamic shape workloads with MLIR. 3 Overview of DISC Figure1describes … scs settees for sale