Searched refs:cache_line_aligned_task_unit (Results 1 – 3 of 3) sorted by relevance
159 constexpr int64_t cache_line_aligned_task_unit = cache_line_size / sizeof(scalar_t); in adagrad_fused_step_impl() local160 size_t num_units = divup(param.numel(), cache_line_aligned_task_unit); in adagrad_fused_step_impl()164 begin *= cache_line_aligned_task_unit; in adagrad_fused_step_impl()165 end = std::min(end * cache_line_aligned_task_unit, param.numel()); in adagrad_fused_step_impl()
204 constexpr int64_t cache_line_aligned_task_unit = cache_line_size / sizeof(scalar_t); in sgd_fused_step_impl() local205 size_t num_units = divup(param.numel(), cache_line_aligned_task_unit); in sgd_fused_step_impl()209 begin *= cache_line_aligned_task_unit; in sgd_fused_step_impl()210 end = std::min(end * cache_line_aligned_task_unit, param.numel()); in sgd_fused_step_impl()
298 constexpr int64_t cache_line_aligned_task_unit = cache_line_size / sizeof(scalar_t); in adam_fused_step_impl() local299 size_t num_units = divup(param.numel(), cache_line_aligned_task_unit); in adam_fused_step_impl()303 begin *= cache_line_aligned_task_unit; in adam_fused_step_impl()304 end = std::min(end * cache_line_aligned_task_unit, param.numel()); in adam_fused_step_impl()