yaml: fix Marlin AssertionError
Marlin quantized linear only supports GPU device, when change generate_op to "KLinearMarlin", generate_device need to be changed to "cuda" accordingly. Fixes: e5b001d7 ("Update readme; Format code; Add example yaml.")
Showing
Please register or sign in to comment