proxy_fw.cpp

#include "src/cuda/multi_head_attn/proxy_fw.h"
#include "megdnn/basic_types.h"
#include "megdnn/dtype.h"
#include "src/cuda/matrix_mul/opr_impl.h"

namespace megdnn {
namespace cuda {

#define cb(DType)                                                                   \
    void MHAForwardProxyOpr::move_scaler_to_device(                                 \
            Handle* handle, DTypeTrait<DType>::ctype* dst,                          \
            DTypeTrait<DType>::ctype* src) {                                        \
        cudaMemcpyAsync(                                                            \
                dst, src, sizeof(DTypeTrait<DType>::ctype), cudaMemcpyHostToDevice, \
                cuda_stream(handle));                                               \
    };
MEGDNN_FOREACH_COMPUTING_DTYPE(cb)
#undef cb

}  // namespace cuda
}  // namespace megdnn
// vim: syntax=cpp.doxygen