Support uint dtype of constant op (#5396)
* refine and add test case * support ellipsis type slice * refine * refine * support slice assign ellipsis type * refine * register fn to localtensor * register uint8 dtype of constantop * auto format by CI Co-authored-by:oneflow-ci-bot <69100618+oneflow-ci-bot@users.noreply.github.com> Co-authored-by:
oneflow-ci-bot <ci-bot@oneflow.org>
Showing
- oneflow/core/kernel/util/cuda_arithemetic_interface.cu 1 addition, 0 deletionsoneflow/core/kernel/util/cuda_arithemetic_interface.cu
- oneflow/core/kernel/util/cuda_arithemetic_interface.h 1 addition, 0 deletionsoneflow/core/kernel/util/cuda_arithemetic_interface.h
- oneflow/core/kernel/util/host_arithemetic_interface.cpp 1 addition, 0 deletionsoneflow/core/kernel/util/host_arithemetic_interface.cpp
- oneflow/core/kernel/util/host_arithemetic_interface.h 1 addition, 0 deletionsoneflow/core/kernel/util/host_arithemetic_interface.h
- oneflow/python/test/modules/test_constant.py 10 additions, 4 deletionsoneflow/python/test/modules/test_constant.py
- oneflow/user/kernels/constant_kernel.cpp 6 additions, 2 deletionsoneflow/user/kernels/constant_kernel.cpp
Please register or sign in to comment