Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

Add a test for map_location="cpu" #497

Merged
merged 1 commit into from
Jul 11, 2024
Merged

Add a test for map_location="cpu" #497

merged 1 commit into from
Jul 11, 2024

Commits on Jul 10, 2024

  1. Add a test for map_location="cpu" 

    Summary:
torchtune is using torch.load(file_name, map_location="cpu", mmap=True), so we add a test
to make sure this works with tensor subclass API

Test Plan:
python test/quantization/test_quant_api.py -k test_quantized_tensor_subclass_save_load_map_location

Reviewers:

Subscribers:

Tasks:

Tags:
    @jerryzh168
    jerryzh168 committed Jul 10, 2024
    Configuration menu
    Copy the full SHA
    9da3032 View commit details
    Browse the repository at this point in the history