Skip to content

GitLab

Commit 16a3bdf3 authored by Michael Carilli's avatar Michael Carilli
Browse files

Updating TensorList->TensorListMetadata

parent 651150cb
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
csrc/multi_tensor_sgd_kernel.cu
View file @ 16a3bdf3
...@@ -31,7 +31,7 @@ struct SGDFunctor ...@@ -31,7 +31,7 @@ struct SGDFunctor
__device__ __forceinline__ void operator()( __device__ __forceinline__ void operator()(
int chunk_size, int chunk_size,
volatile int* noop_gmem, volatile int* noop_gmem,
TensorList<N>& tl, TensorListMetadata<N>& tl,
float wd, float wd,
float momentum, float momentum,
float dampening, float dampening,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment