aparler/c-llm
1
0
Fork 0
mirror of https://github.com/karpathy/llm.c.git synced 2026-07-26 20:15:08 -04:00
Code Issues Projects Releases Packages Wiki Activity Actions

Merge pull request #514 from ngc92/compile-fix

Browse source 
fix compilation with older nvcc
This commit is contained in:
Andrej 2024-06-02 16:42:31 -07:00 • committed by GitHub
commit cb5eff0578
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 2 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

4
train_gpt2.cu
View file

@ -228,10 +228,10 @@ struct alignas(16) Packed128 {
return result;
}
__device__ static Packed128 zeros() {
return constant(0);
return constant(0.f);
}
__device__ static Packed128 ones() {
return constant(1);
return constant(1.f);
}
__device__ ElementType& operator[](int index) {