Metadata-Version: 2.2
Name: mlsysutil
Version: 0.1.0
Summary: A package for AI system Utils
Home-page: https://github.com/jelite/mlsysUtils
Author: texnee
Author-email: texnee@gmail.com
Requires-Python: >=3.6
Description-Content-Type: text/markdown
Dynamic: author
Dynamic: author-email
Dynamic: home-page
Dynamic: requires-python

# MyPackage

This package provides functionality for CUDA Graph capturing in PyTorch. It includes a warm-up function and CUDA Graph capture function to optimize inference execution.

## Installation

To install this package, run the following command:

```sh
pip install mlsysUtil
```

Or install from the local build:

```sh
pip install dist/mlsysUtils-0.1.0-py3-none-any.whl
```

## Usage

```python
import torch
from mlsysUtils.graph_capture import get_graph

model = torch.nn.Linear(10, 10).cuda()
input_tensor = torch.randn(1, 10).cuda()

graph, static_output = get_graph(model, input_tensor)
```

## License

This project is licensed under the MIT License.

