Skip to content

Commit

Permalink
minor fix
Browse files Browse the repository at this point in the history
  • Loading branch information
garrett4wade committed Feb 24, 2024
1 parent f1c1490 commit 0213d23
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion csrc/adam/multi_tensor_adam.cu
Original file line number Diff line number Diff line change
Expand Up @@ -23,7 +23,7 @@ This file is adapted from fused adam in NVIDIA/apex, commit a109f85
#define BLOCK_SIZE 512
#define ILP 4

typedef enum: int {
typedef enum : int {
ADAM_MODE_0 = 0, // L2 regularization mode
ADAM_MODE_1 = 1 // Decoupled weight decay mode(AdamW)
} adamMode_t;
Expand Down

0 comments on commit 0213d23

Please sign in to comment.