SCIInstitute / shapeworks-cloud

A web version of ShapeWorks Studio
https://www.shapeworks-cloud.org/#/
Apache License 2.0
1 stars 0 forks source link

Check gpu task #361

Closed annehaley closed 5 months ago

annehaley commented 5 months ago

This PR is intended to demonstrate that the mock deepssm task has access to a GPU when run from the "gpu" queue. These changes can be reverted when the real deepssm task is implemented, but for now this version can be used to test the GPU availability.

In order to demonstrate a difference between the two queues, the "mock-deepssm" endpoint will spawn a task on each queue. To persist the feedback from each task, two TaskProgress objects are used. These TaskProgress objects are not associated with any project, so one migration needs to be applied to allow project=null on TaskProgress objects. Since we need a migration anyway, I added a field "message" to the model (similar to the "error" field but without the connotation of failure). The mock deepssm task will save a string to this field which describes the availability of a GPU device.

Once merged, the expected behavior is as follows: