Updated to use kp instead of komputepy for module name

This commit is contained in:
Alejandro Saucedo
2020-11-02 21:57:10 +00:00
parent b23e04e1a4
commit 991cfdcbcc
5 changed files with 11 additions and 11 deletions

View File

@@ -5,7 +5,7 @@
namespace py = pybind11;
PYBIND11_MODULE(komputepy, m) {
PYBIND11_MODULE(kp, m) {
py::enum_<kp::Tensor::TensorTypes>(m, "TensorTypes", "Enum with GPU memory types for Tensor.")
.value("device", kp::Tensor::TensorTypes::eDevice, "Tensor holding data in GPU memory.")

View File

@@ -1,5 +1,5 @@
from komputepy import Tensor, Manager, Sequence
from kp import Tensor, Manager, Sequence
def test_opmult():
"""