Skip to content

Commit

Permalink
[Typo]Fix a typo for index. (pytorch#106447)
Browse files Browse the repository at this point in the history
Fixes #ISSUE_NUMBER

Pull Request resolved: pytorch#106447
Approved by: https://github.com/awgu
  • Loading branch information
summerdo authored and pytorchmergebot committed Aug 3, 2023
1 parent a6f7dd4 commit 6e2a284
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion aten/src/ATen/native/cpu/IndexKernelUtils.h
Original file line number Diff line number Diff line change
Expand Up @@ -57,7 +57,7 @@ void cpu_index_kernel(TensorIteratorBase& iter, IntArrayRef index_size, IntArray
const func_t& f, bool serial_execution=false)
{
int ntensor = iter.ntensors();
// When launch the index parallel version, set a relative samll grain size less than the INTERNAL::GRAIN_SIZE
// When launch the index parallel version, set a relative small grain size less than the INTERNAL::GRAIN_SIZE
// to make the whole available thread numbers get more balanced work load and a better cache location.
// The grain size here is chosen by the op benchmark to overcome the thread launch overhead
const int index_parallel_grain_size = 3000;
Expand Down

0 comments on commit 6e2a284

Please sign in to comment.