DeepLink-org / deeplink.framework

BSD 3-Clause "New" or "Revised" License
55 stars 28 forks source link

feat(dipu): set pin_memory to false for unsupported vendor #898

Closed Gong-air closed 1 month ago

Gong-air commented 1 month ago

S2等没有实现pin_memory的vendor,需要在框架层对pin_memory设置为False,是不方便的。 因此在DIPU层对vendor进行判断,强行设置pin_memory为False,并打印相关LOG。