h2oai / h2o4gpu

H2Oai GPU Edition
Apache License 2.0
460 stars 95 forks source link

Use less memory for prediction #744

Open sh1ng opened 5 years ago

sh1ng commented 5 years ago

Ideally it should running also on GPU. But even without it naive implementation https://github.com/h2oai/h2o4gpu/blob/master/src/interface_py/h2o4gpu/solvers/factorization.py#L309-L311 can be optimized to use less memory.