if(WITH_GPU OR WITH_ROCM)
  cc_library(
    cuda_stream
    SRCS cuda_stream.cc
    DEPS enforce eigen3 ${MKLDNN_CTX_DEPS})
endif()