"torch_harmonics/cache.py" did not exist on "55bbcb251355cdedc59a4bccc6d6c38d83512462"
  • 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