Code

Add MaskedL2NN#838

Code

Merged

rapids-bot[bot] merged 57 commits intorapidsai:branch-23.02rapidsai/raft:branch-23.02from ahendriksen:enh-sparse-l2-nnahendriksen/raft:enh-sparse-l2-nnCopy head branch name to clipboardJan 27, 2023

+1,923-4

Commits

Commits on Jan 24, 2023

• contractions: Concentrate tile index calculations

  Show description for a3b8587
  

  ahendriksen

  authored and

  Allard Hendriksen

  committed
  a3b8587

• pairwise_distance_base: Remove all ldgXY(0) calls

  Show description for 99e65a5
  

  ahendriksen

  authored and

  Allard Hendriksen

  committed
  99e65a5

• pairwise_distance_base: Move all logic into run loop

  Show description for e6d5078
  

  ahendriksen

  authored and

  Allard Hendriksen

  committed
  e6d5078

• pairwise_distance_base: Fix typo

  Show description for 995d2ae
  

  ahendriksen

  authored and

  Allard Hendriksen

  committed
  995d2ae

• Implement reviewer feedback

  

  ahendriksen

  authored and

  Allard Hendriksen

  committed
  e6976c5

Commits on Jan 25, 2023

• Merge branch 'branch-23.02' into wip-move-contractions-tiling-logic

  cjnolet

  authored
  4947dc8

• Add sparseL2NN initial implementation

  ahendriksen

  committed
  8385f2f

• Add sparseL2NN benchmarks

  ahendriksen

  committed
  69687d6

• Rename files sparse_* => masked_*

  ahendriksen

  committed
  e74b5f4

• Rename functions sparse_* => masked_*

  Show description for 7bf5801
  ahendriksen

  committed
  7bf5801

• Remove workspace parameter

  ahendriksen

  committed
  96ff79d

• Use mdspan parameters

  ahendriksen

  committed
  be56826

• Remove uvectors from benchmark

  Show description for 07c2f53
  ahendriksen

  committed
  07c2f53

• Use parameter struct in public API

  ahendriksen

  committed
  2f23481

• Clean up minor nits in tests and benchmarks

  ahendriksen

  committed
  7df1fd2

• Update copyright years

  ahendriksen

  committed
  b590684

• Rename masked_l2_nn => masked_nn

  ahendriksen

  committed
  9fcc24d

• cmake: rename masked_l2_nn => masked_nn

  ahendriksen

  committed
  953b33d

• Clang-format

  ahendriksen

  committed
  724e874

• Fix docs for masked NN

  ahendriksen

  committed
  4ea41c4

• Reduce iterations of deterministic test

  Show description for 529764e
  ahendriksen

  committed
  529764e

• Add SDDM comparison

  ahendriksen

  committed
  f4db5e5

• Fix const and add extents checks

  ahendriksen

  committed
  b0954e8

• Refactor maskedL2NN test

  Show description for 2e3ca44
  ahendriksen

  committed
  2e3ca44

• wording: grouped -> processed

  ahendriksen

  committed
  db6288c

• Docstring changes and removal of extraneous function

  ahendriksen

  committed
  773ab2f

• Move sqrt from template to run-time parameter

  Show description for bae2bcb
  ahendriksen

  committed
  bae2bcb

• Remove extraneous memcpy::async stuff

  ahendriksen

  committed
  0629446

• Implement half the reviewer feedback on mnn_base

  ahendriksen

  committed
  33787ce

• Formatting

  ahendriksen

  committed
  6243c8a

• Fix copyright years

  Show description for 4dc0c37
  ahendriksen

  committed
  4dc0c37

• Reword comment

  ahendriksen

  committed
  b84a5b4

• test: Remove redundant comparison operator

  ahendriksen

  committed
  7fb7cc9

• Add sphinx docsfor maskedL2NNImpl

  ahendriksen

  committed
  fe31a13

• Document thread_adj

  ahendriksen

  committed
  3bf204f

• Rename tile_end_n -> group_end_n

  Show description for 08448d8
  ahendriksen

  committed
  08448d8

• Formatting

  ahendriksen

  committed
  ed178ca

• Add compress_to_bits kernel wrapper

  ahendriksen

  committed
  80cb3b0

• Add compress_to_bits docs and test

  ahendriksen

  committed
  546854d

• masked_nn: Fix test param printer

  ahendriksen

  committed
  125ae98

• Use mdspan for compress_to_bits

  Show description for 6e77653
  ahendriksen

  committed
  6e77653

• Make benchmark more informative

  Show description for 8925146
  ahendriksen

  committed
  8925146

• Merge branch 'branch-23.02' into enh-sparse-l2-nn

  cjnolet

  authored
  3e86024

• Merge branch 'branch-23.02' into wip-move-contractions-tiling-logic

  cjnolet

  authored
  ba6491a

Commits on Jan 26, 2023

• Forcing sccache reinit.

  cjnolet

  committed
  e52b0f9

• Merge branch 'branch-23.02' into wip-move-contractions-tiling-logic

  cjnolet

  committed
  34eb76a

• Breaking specializations for refine into individual files

  cjnolet

  committed
  85c6294

• Checking in

  cjnolet

  committed
  0fad842

• Including just the refine specialization

  cjnolet

  committed
  f7788af

• Merge branch 'branch-23.02' into wip-move-contractions-tiling-logic

  cjnolet

  authored
  e626101

• Proper import of speicalizations

  cjnolet

  committed
  9e7b729

• Merge branch 'wip-move-contractions-tiling-logic' of github.com:ahendriksen/raft into wip-move-contractions-tiling-logic

  cjnolet

  committed
  9e4b5f3

• Remove SCCACHE_RECACHE from build.sh

  cjnolet

  committed
  060e62c

• Merge branch 'wip-move-contractions-tiling-logic' into enh-sparse-l2-nn

  cjnolet

  committed
  2870b67

• Small compilation error remains

  cjnolet

  committed
  2b0c02b

Commits on Jan 27, 2023

• Take device_resources instead of handle

  ahendriksen

  committed
  1e83640

• Merge remote-tracking branch 'rapids/branch-23.02' into enh-sparse-l2-nn

  ahendriksen

  committed
  862e8f6