• 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_subgraph.py 27.3 KB