diff options
| author | Tim Dettmers <dettmers@cs.washington.edu> | 2021-10-20 19:15:47 -0700 |
|---|---|---|
| committer | Tim Dettmers <dettmers@cs.washington.edu> | 2021-10-20 19:15:47 -0700 |
| commit | a6eae2e7f2bf03f268fcb6b055201ff6827684c4 (patch) | |
| tree | d2f72792251c9feaef1cf9dcddc3c79e6312a93a /csrc/pythonInterface.c | |
| parent | bb34fd50a1fec74e62beb6e23d51f0142c7d0ab6 (diff) | |
Added skip_zeros; tests are passing.
Diffstat (limited to 'csrc/pythonInterface.c')
| -rw-r--r-- | csrc/pythonInterface.c | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/csrc/pythonInterface.c b/csrc/pythonInterface.c index 67bf2e5..7d5e654 100644 --- a/csrc/pythonInterface.c +++ b/csrc/pythonInterface.c @@ -110,7 +110,7 @@ extern "C" float eps, int step, float lr, \ float* quantiles1, float* quantiles2, \ float* max1, float* max2, float* new_max1, float* new_max2, \ - float weight_decay, float gnorm_scale, bool skip_zeros, int n) \ + float weight_decay, float gnorm_scale, int n) \ { \ name##_static_8bit_g##gbits(g, p, state1, state2, unorm, max_unorm, param_norm, beta1, beta2, eps, step, lr, \ quantiles1, quantiles2, max1, max2, new_max1, new_max2, weight_decay, gnorm_scale, n); \ |