You can not select more than 25 topics Topics must start with a chinese character,a letter or number, can include dashes ('-') and can be up to 35 characters long.

get_params.template 1.0 kB

12345678910111213141516171819202122232425262728293031
  1. #include "src/cuda/convolution/cudnn_heuristic.h"
  2. #include "megdnn.h"
  3. using namespace megdnn;
  4. using namespace cuda;
  5. using namespace convolution;
  6. bool convolution::heuristic_params_available(
  7. int cuda_major, int cuda_minor, size_t* layer_num_p,
  8. const size_t** layers_dim_p, const float** matrices_p,
  9. const float** biases_p, const float** alpha_p, const float** beta_p,
  10. const ConvolutionType& conv_type, float** hidden_units_p,
  11. float** time_pred_p, float** mask_p) {
  12. MEGDNN_MARK_USED_VAR(cuda_major);
  13. MEGDNN_MARK_USED_VAR(cuda_minor);
  14. MEGDNN_MARK_USED_VAR(layer_num_p);
  15. MEGDNN_MARK_USED_VAR(layers_dim_p);
  16. MEGDNN_MARK_USED_VAR(matrices_p);
  17. MEGDNN_MARK_USED_VAR(biases_p);
  18. MEGDNN_MARK_USED_VAR(alpha_p);
  19. MEGDNN_MARK_USED_VAR(beta_p);
  20. MEGDNN_MARK_USED_VAR(conv_type);
  21. MEGDNN_MARK_USED_VAR(hidden_units_p);
  22. MEGDNN_MARK_USED_VAR(time_pred_p);
  23. MEGDNN_MARK_USED_VAR(mask_p);
  24. {cudnn_select}
  25. return false;
  26. }
  27. // vim: syntax=cpp.doxygen

MegEngine 安装包中集成了使用 GPU 运行代码所需的 CUDA 环境,不用区分 CPU 和 GPU 版。 如果想要运行 GPU 程序,请确保机器本身配有 GPU 硬件设备并安装好驱动。 如果你想体验在云端 GPU 算力平台进行深度学习开发的感觉,欢迎访问 MegStudio 平台

Contributors (1)