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