CINN icon indicating copy to clipboard operation
CINN copied to clipboard

fix cudnn multi thread safe

Open SunNy820828449 opened this issue 3 years ago • 0 comments

paddle gpu计算使用了相同的计算流,因此可以共享使用同一个workspace。pr使用了shared_ptr管理共享的workspace,能够实现智能的指针管理。由于不同计算需要workspace大小不一样,因此会需要修改workspace的空间,子线程解决智能指针的释放问题,并通过event进行同步。

SunNy820828449 avatar Nov 13 '21 07:11 SunNy820828449