Commit 456aaa2f authored by Wen-Heng (Jack) Chung's avatar Wen-Heng (Jack) Chung
Browse files

[ROCm] Interface changes for pooling APIs in StreamExecutor

Due to the design of MIOpen, the DNN library on ROCm platform, an instance of
ScratchAllocator has to be passed into pooling routines. This commit address
such interface changes and the implementation in CUDA StreamExecutor.
parent 135e419e
Loading
Loading
Loading
Loading
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please to comment