• Paul Fultz II's avatar
    Handle removing contiguous on operators that use modules (#1005) · 785307c3
    Paul Fultz II authored
    Currently, eliminate_contiguous will never remove contiguous for operators that use module inputs due to the fact that it doesn't pass the module inputs to compute_shape.
    
    - Update to pass the module inputs correctly to compute_shape
    - Fix the overloads of compute_shape so that when passed an empty vector of module inputs it will call the overload without module inputs
    - Add tests with contiguous and pointwise module function.
    - Move add_pointwise function to a seperate header to reuse across different tests
    785307c3
adjust_allocation.cpp 3.62 KB