Skip to content

README.md

Latest commit

 

History

History
51 lines (36 loc) · 1001 Bytes

README.md

File metadata and controls

51 lines (36 loc) · 1001 Bytes
Raw

mcfile-python

A basic Python wrapper for the MACA mcFile API.

This repo is forked from https://github.com/yanok/cufile-python.git

Installation

pip install mcfile-python

Usage

Basic usage with the CuFile context manager:

from mcfile import CuFile
import torch
import ctypes

# allocate an empty buffer in VRAM
t = torch.empty((1024, 1024, 16), dtype=torch.float32, device="cuda")
with CuFile("test.bin", "r") as f:
    f.read(ctypes.c_void_p(t.data_ptr()), t.nbytes)

Alternatively one could import mcfile.bindings and use it as if calling libmcfile from C++.

Development

To set up the development environment:

  1. Clone the repository:
git clone https://github.com/MetaX-MACA/mcfile-python.git
cd mcfile-python
  1. Install development dependencies:
pip install -r requirements.txt
  1. Run tests:
python -m pytest tests/

License

This project is licensed under the MIT License - see the LICENSE file for details.