Skip to content
This repository has been archived by the owner on Nov 17, 2023. It is now read-only.

Add boolean ndarray #15940

Merged
merged 4 commits into from
Oct 8, 2019
Merged

Add boolean ndarray #15940

merged 4 commits into from
Oct 8, 2019

Commits on Oct 8, 2019

  1. Initial infra of boolean ndarray 

    Add np.equal implemented using tvmop

Fix setting DLDataType conversion for boolean ndarray

Add equal_gpu

Fix inputs with different ndims

Fix copying boolean ndarrays across devices

Refactor binary logic op impl by tvm

Add more logic ops

Refactor TVMOpModule::Call to CallEx

Add binary scalar logic op expr and schedule

Add binary scalar logic ops

Add free functions for logic ops

Rebase with master to fix SetDLTensor bug

Fix pylint

Add sum op for boolean ndarrays using tvm op module

Add sum boolean gpu compute

Add bool type support to boolean_mask

Boolean indexing working

Clean up

Fix merge

Sync Makefile

Rebase

Add boolean indexing test

Fix sanity

Fix gpu and add autograd test

Rebase

Fix test for windows

Fix tests

Try to fix cuda arch missing error in ci

Fix ci

Fix windows build

Try to fix cmake

Fix cmake

Fix

Revert config.mk
    @reminisce
    reminisce committed Oct 8, 2019
    Configuration menu
    Copy the full SHA
    57360cd View commit details
    Browse the repository at this point in the history

  2. Fix cmake

    @reminisce
    reminisce committed Oct 8, 2019
    Configuration menu
    Copy the full SHA
    9a04649 View commit details
    Browse the repository at this point in the history

  3. Skip compute capability <= 52 for TVM generated ops

    @reminisce
    reminisce committed Oct 8, 2019
    Configuration menu
    Copy the full SHA
    0911be4 View commit details
    Browse the repository at this point in the history

  4. Fix sanity

    @reminisce
    reminisce committed Oct 8, 2019
    Configuration menu
    Copy the full SHA
    a6dac14 View commit details
    Browse the repository at this point in the history