FTorch icon indicating copy to clipboard operation
FTorch copied to clipboard
verified publisher icon Cambridge-ICCS

GPU demo

Open joewallwork opened this issue 1 year ago • 0 comments

This PR includes various tweaks in preparation for giving a GPU demo:

  • Provides templates for switching to GPU across the examples.
  • Fixes some Python imports in example 3.
  • Compiles example 3 if CUDA is enabled.
  • Some naming consistency.

We should follow this up by adding tests for example 3. I propose doing this when addressing #142.

[The issue I was having was due to a bug I accidentally introduced, which I've now fixed.]

joewallwork avatar Jul 05 '24 14:07 joewallwork