"tests/quantization/torchao/__init__.py" did not exist on "d8287fcd1d94f33df55b54e2e1c140c2ab15b444"
  • nv-dlasalle's avatar
    Prevent users from attempting to pin PyTorch non-contiguous tensors or views... · 1f2e6960
    nv-dlasalle authored
    
    Prevent users from attempting to pin PyTorch non-contiguous tensors or views only encompassing part of tensor. (#3992)
    
    * Disable pinning non-contiguous memory
    
    * Prevent views from being converted for write
    
    * Fix linting
    
    * Add unit tests
    
    * Improve error message for users
    
    * Switch to pytest function
    
    * exclude mxnet and tensorflow from inplace pinning
    
    * Add skip
    
    * Restrict to pytorch backend
    
    * Use backend to retrieve device
    
    * Fix capitalization in decorator
    Co-authored-by: default avatarQuan (Andy) Gan <coin2028@hotmail.com>
    1f2e6960
test_pin_memory.py 738 Bytes