Saved in:
| Main Authors: | , , , , , |
|---|---|
| Format: | Preprint |
| Published: |
2024
|
| Subjects: | |
| Online Access: | https://arxiv.org/abs/2410.03210 |
| Tags: |
Add Tag
No Tags, Be the first to tag this record!
|
| _version_ | 1866909630763892736 |
|---|---|
| author | Vatai, Emil Drozd, Aleksandr Ivanov, Ivan R. Batista, Joao E. Ren, Yinghao Wahib, Mohamed |
| author_facet | Vatai, Emil Drozd, Aleksandr Ivanov, Ivan R. Batista, Joao E. Ren, Yinghao Wahib, Mohamed |
| contents | Frameworks and domain-specific languages for auto-generating code have traditionally depended on human experts to implement rigorous methods ensuring the legality of code transformations. Recently, machine learning (ML) has gained traction for generating code optimized for specific hardware targets. However, ML approaches-particularly black-box neural networks-offer no guarantees on the correctness or legality of the transformations they produce. To address this gap, we introduce Tadashi, an end-to-end system that leverages the polyhedral model to support researchers in curating datasets critical for ML-based code generation. Tadashi provides an end-to-end system capable of applying, verifying, and evaluating candidate transformations on polyhedral schedules with both reliability and practicality. We formally prove that Tadashi guarantees the legality of generated transformations, demonstrate its low runtime overhead, and showcase its broad applicability. Tadashi available at https://github.com/vatai/tadashi/. |
| format | Preprint |
| id |
arxiv_https___arxiv_org_abs_2410_03210 |
| institution | arXiv |
| publishDate | 2024 |
| record_format | arxiv |
| spellingShingle | Tadashi: Enabling AI-Based Automated Code Generation With Guaranteed Correctness Vatai, Emil Drozd, Aleksandr Ivanov, Ivan R. Batista, Joao E. Ren, Yinghao Wahib, Mohamed Machine Learning Frameworks and domain-specific languages for auto-generating code have traditionally depended on human experts to implement rigorous methods ensuring the legality of code transformations. Recently, machine learning (ML) has gained traction for generating code optimized for specific hardware targets. However, ML approaches-particularly black-box neural networks-offer no guarantees on the correctness or legality of the transformations they produce. To address this gap, we introduce Tadashi, an end-to-end system that leverages the polyhedral model to support researchers in curating datasets critical for ML-based code generation. Tadashi provides an end-to-end system capable of applying, verifying, and evaluating candidate transformations on polyhedral schedules with both reliability and practicality. We formally prove that Tadashi guarantees the legality of generated transformations, demonstrate its low runtime overhead, and showcase its broad applicability. Tadashi available at https://github.com/vatai/tadashi/. |
| title | Tadashi: Enabling AI-Based Automated Code Generation With Guaranteed Correctness |
| topic | Machine Learning |
| url | https://arxiv.org/abs/2410.03210 |