src/array/cpu/spmm_binary_ops.h · a7e941c379f0f9ab472c844b6a9f1d05d687b4e1 · OpenDAS / dgl

"git@developer.sourcefind.cn:change/sglang.git" did not exist on "98eecbda54d5f3f90ba984bbc450cd15225450bd"

[Performance] Dynamic cpu kernel V3 for SpMMSumCsr all Ops (#2309) · f8ebcd7f

pawelpiotrowicz authored Nov 17, 2020



* support AVX512

* env DGL_CPU_INTEL_KERNEL_ENABLED=1

* env DGL_CPU_INTEL_KERNEL_LOG=1

* Add unittest test_spmm.cc
Co-authored-by: Izabela Mazur <izabela.mazur@intel.com>
Co-authored-by: Michal Szarmach <michal.szarmach@intel.com>

Review patch

f8ebcd7f

spmm_binary_ops.h 4.94 KB

Replace spmm_binary_ops.h