1. 12 Aug, 2026 1 commit
    • 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
  2. 10 Aug, 2026 3 commits