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.

acl.h 1.2 kB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950
  1. /**
  2. * @file acl.h
  3. *
  4. * Copyright (C) Huawei Technologies Co., Ltd. 2019-2020. All Rights Reserved.
  5. *
  6. * This program is distributed in the hope that it will be useful,
  7. * but WITHOUT ANY WARRANTY; without even the implied warranty of
  8. * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
  9. */
  10. #ifndef INC_EXTERNAL_ACL_ACL_H_
  11. #define INC_EXTERNAL_ACL_ACL_H_
  12. #include "acl_rt.h"
  13. #include "acl_op.h"
  14. #include "acl_mdl.h"
  15. #ifdef __cplusplus
  16. extern "C" {
  17. #endif
  18. /**
  19. * @ingroup AscendCL
  20. * @brief acl initialize
  21. *
  22. * @par Restriction
  23. * The aclInit interface can be called only once in a process
  24. * @param configPath [IN] the config path,it can be NULL
  25. * @retval ACL_ERROR_NONE The function is successfully executed.
  26. * @retval OtherValues Failure
  27. */
  28. ACL_FUNC_VISIBILITY aclError aclInit(const char *configPath);
  29. /**
  30. * @ingroup AscendCL
  31. * @brief acl finalize
  32. *
  33. * @par Restriction
  34. * Need to call aclFinalize before the process exits.
  35. * After calling aclFinalize,the services cannot continue to be used normally.
  36. * @retval ACL_ERROR_NONE The function is successfully executed.
  37. * @retval OtherValues Failure
  38. */
  39. ACL_FUNC_VISIBILITY aclError aclFinalize();
  40. #ifdef __cplusplus
  41. }
  42. #endif
  43. #endif // INC_EXTERNAL_ACL_ACL_H_

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