https://github.com/exanauts/exatron.jl
Julia implementation of TRON solver on GPUs
https://github.com/exanauts/exatron.jl
admm algorithm batch gpus
Last synced: 3 months ago
JSON representation
Julia implementation of TRON solver on GPUs
- Host: GitHub
- URL: https://github.com/exanauts/exatron.jl
- Owner: exanauts
- License: mit
- Created: 2020-10-30T21:35:06.000Z (over 4 years ago)
- Default Branch: main
- Last Pushed: 2023-09-25T13:10:22.000Z (over 1 year ago)
- Last Synced: 2024-05-09T23:20:28.558Z (about 1 year ago)
- Topics: admm, algorithm, batch, gpus
- Language: Julia
- Homepage:
- Size: 10.2 MB
- Stars: 15
- Watchers: 8
- Forks: 4
- Open Issues: 4
-
Metadata Files:
- Readme: README.md
- License: LICENSE
- Citation: CITATION.cff
Awesome Lists containing this project
README
# ExaTron.jl
[![][build-release-img]][build-url]
[![DOI][doi-img]][doi-url]
ExaTron.jl implements a trust-region Newton solver for batched nonlinear programming on GPUs.
Problems in the batch are solved in parallel by employing multiple thread blocks on GPUs.
Our basic algorithm to solve each problem on GPUs is an extension of the
algorithm by [Lin and More](https://doi.org/10.1137/S1052623498345075)
and its code [TRON](https://www.mcs.anl.gov/~more/tron).
Please check out [this branch](https://github.com/exanauts/ExaTron.jl/tree/sc2021) to see how to generate PTX code for our kernel and reproduce the experimental results in our [technical report](https://arxiv.org/abs/2106.14995).
## Installation
```julia
pkg> add ExaTron
```
## Citing this package
```
@misc{ExaTron.jl.0.0.0,
author = {Kim, Youngdae and Pacaud, Fran\ccois and Kim, Kibaek},
title = {{ExaTron.jl: GPU-capable TRON solver in Julia}},
month = Mar,
year = 2021,
version = {0.0.0},
url = {https://github.com/exanauts/ExaTron.jl}
}
```
## Acknowledgements
This material is based upon work supported by the U.S. Department of Energy, Office of Science, under contract number DE-AC02-06CH11357.
[build-release-img]: https://github.com/exanauts/ExaTron.jl/workflows/Run%20tests/badge.svg?branch=main
[build-url]: https://github.com/exanauts/ExaTron.jl/actions?query=workflow
[doi-img]: https://zenodo.org/badge/DOI/10.5281/zenodo.5829757.svg
[doi-url]: https://doi.org/10.5281/zenodo.5829757