• shivadbhavsar's avatar
    pybind updates for torch_migraphx library (#1323) · 8045f7c8
    shivadbhavsar authored
    Add function argument_from_pointer to allow directly passing a migraphx.shape object and a memory address. 
    Expose the is_compiled() method from migraphx::program. 
    Expose the enum types under migraphx::op. 
    8045f7c8
migraphx_py.cpp 18 KB