Skip to content

[Model] Specialize model execution devices over CPU and CUDA#89

Merged
ganler merged 4 commits intoise-uiuc:mainfrom
ganler:fix-half-gpu
Mar 7, 2023
Merged

[Model] Specialize model execution devices over CPU and CUDA#89
ganler merged 4 commits intoise-uiuc:mainfrom
ganler:fix-half-gpu

Conversation

@ganler
Copy link
Member

@ganler ganler commented Mar 7, 2023

fixes: #84

Specialize eager mode execution over CPU/CUDA according to backend.target. That being said, if setting backend.target=cuda, the TF/PT model will run over GPUs as well (e.g., TorchModelCUDA) in the eager model (when generating oracles & refining weights).

Also, test cases are split into finer-grains to resolve dependency conflicts brought by recent updates of SUTs.

@ganler ganler merged commit e2da5cb into ise-uiuc:main Mar 7, 2023
@ganler ganler mentioned this pull request Mar 8, 2023
1 task
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

🐛 [BUG] Half precision kernels not implemented for CPU

1 participant