Synchronization on accelerators different than cuda or cpu may not work

cezbloch / pytorchures

Python profiler for pytorch models. Wraps each layer and measures it's execution time.

MIT License

2 stars 0 forks source link

Synchronization on accelerators different than cuda or cpu may not work #11

Closed cezbloch closed 1 month ago

cezbloch commented 1 month ago

In forward call in TimedLayer there is an 'if-statement' checking if cuda is available then synchronize. This obviously works only for cuda and not other accelerators. Try adding other accelerators and synchronize if needed.