Cooperative thread array granularity context switch during trap handling

A non-transitory computer readable storage medium comprising instructions that cause a processor to restore threads within a processing core, by performing the steps of:

• for a first thread group included in a plurality of thread groups, executing a context restore routine to restore from a memory a first portion of a context associated with the first thread group;
• determining whether the first thread group completed an assigned function;
• and
• if the first thread group completed the assigned function, then exiting the context restore routine, or
• if the first thread group did not complete the assigned function, then executing one or more operations associated with a trap handler routine.

A computer system comprising:

• a memory that stores instructions and data related to a trap handling routine;
• a processor that executes the instructions to restore threads within a processing core by performing the steps of:
• for each thread group included in a plurality of thread groups, executing a context restore routine to restore from a memory a first portion of a ontext associated with the thread group;
• for each thread group included in the plurality of thread groups, determining whether the thread group completed an assigned function;
• and
• if the thread group completed an assigned function, then exiting the context restore routine, or
• if the thread group did not complete the assigned function, then executing one or more operations associated with a trap handler routine.