mirror of
https://github.com/nomic-ai/kompute.git
synced 2026-05-11 08:59:59 +00:00
Updated to use kp instead of komputepy for module name
This commit is contained in:
@@ -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.")
|
||||
|
||||
@@ -1,5 +1,5 @@
|
||||
|
||||
from komputepy import Tensor, Manager, Sequence
|
||||
from kp import Tensor, Manager, Sequence
|
||||
|
||||
def test_opmult():
|
||||
"""
|
||||
|
||||
Reference in New Issue
Block a user