diff --git a/support/accelerator.go b/support/accelerator.go index 2fb14a4..7e9e78f 100644 --- a/support/accelerator.go +++ b/support/accelerator.go @@ -13,6 +13,6 @@ type Accelerator struct { } // Method to check if the accelerator is a GPU -func (a Accelerator) isGpu() bool { +func (a Accelerator) IsGpu() bool { return a != CPU }