The source code and dockerfile for the GSW2024 AI Lab.
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
This repo is archived. You can view files and clone it, but cannot push or open issues/pull-requests.

11 lines
273 B

4 weeks ago
  1. #ifndef STORM_CUDAFORSTORM_UTILITY_H_
  2. #define STORM_CUDAFORSTORM_UTILITY_H_
  3. // Library exports
  4. #include "cudaForStorm.h"
  5. size_t getFreeCudaMemory();
  6. size_t getTotalCudaMemory();
  7. bool resetCudaDevice();
  8. int getRuntimeCudaVersion();
  9. #endif // STORM_CUDAFORSTORM_UTILITY_H_