Add model training GPU regression test
Open
area: gpu
type: enhancement
- Dominant language
- Go
- Stars
- 19.3k
- Forks
- 2k
- Avg merge
- 3d 5h
- Merged PRs (30d)
- 264
Description
### Description
Our only GPU regression tests are doing inference. We should have a test that does some training too.
### Is this feature related to a specific bug?
#10879
### Do you have a specific solution in mind?
#10879 has an example Dockerfile which is modeled after https://github.com/karpathy/llm.c/discussions/677.
Contributor guide
Assessment
This issue has not been assessed yet.