• xeron56's avatar
    Add CUDA device compatibility checks to TimesFM backend · 6132f5c0
    xeron56 authored
    - Implemented a function to check if the current CUDA device is supported by the installed PyTorch version.
    - Added warnings for unsupported CUDA devices, informing users that the CPU will be used instead.
    - Updated the TimesFMBackend class to utilize the new compatibility checks and handle GPU names appropriately.
    - Introduced unit tests to validate the behavior of the CUDA compatibility function with various simulated CUDA environments.
    6132f5c0
Name
Last commit
Last update
..
design-qa Loading commit data...
src Loading commit data...
.gitignore Loading commit data...
README.md Loading commit data...
angular.json Loading commit data...
package-lock.json Loading commit data...
package.json Loading commit data...
tsconfig.app.json Loading commit data...
tsconfig.json Loading commit data...