• Jeremy Reizenstein's avatar
    pulsar build and CI changes · d220ee2f
    Jeremy Reizenstein authored
    Summary:
    Changes to CI and some minor fixes now that pulsar is part of pytorch3d. Most significantly, add CUB to CI builds.
    
    Make CUB_HOME override the CUB already in cudatoolkit (important for cuda11.0 which uses cub 1.9.9 which pulsar doesn't work well with.
    Make imageio available for testing.
    Lint fixes.
    Fix some test verbosity.
    Avoid use of atomicAdd_block on older GPUs.
    
    Reviewed By: nikhilaravi, classner
    
    Differential Revision: D24773716
    
    fbshipit-source-id: 2428356bb2e62735f2bc0c15cbe4cff35b1b24b8
    d220ee2f
pulsar_optimization_unified.py 6.18 KB