issue/410 Feature: Add infinicore python package

Author: voltjia

.github/workflows/build.yml

@@ -31,7 +31,7 @@ jobs:
         xmake-version: latest
 
     - name: Build & Install
-      run: python scripts/install.py --omp=y
+      run: python scripts/install.py --omp=y -y
 
     - name: install python packages
       run: |

MANIFEST.in

@@ -0,0 +1 @@
+global-include *

include/infinicore.hpp

@@ -0,0 +1,6 @@
+#ifndef __INFINICORE_API_HPP__
+#define __INFINICORE_API_HPP__
+
+#include "infinicore/tensor.hpp"
+
+#endif

include/infinicore/device.hpp

@@ -0,0 +1,37 @@
+#ifndef __INFINICORE_DEVICE_API_HPP__
+#define __INFINICORE_DEVICE_API_HPP__
+
+#include <cstdint>
+#include <string>
+
+namespace infinicore {
+
+class Device {
+public:
+    using Index = std::size_t;
+
+    enum class Type {
+        cpu,
+        cuda,
+        meta,
+    };
+
+    Device(const Type &type, const Index &index = 0);
+
+    const Type &get_type() const;
+
+    const Index &get_index() const;
+
+    std::string to_string() const;
+
+    static std::string to_string(const Type &type);
+
+private:
+    Type type_;
+
+    Index index_;
+};
+
+} // namespace infinicore
+
+#endif

include/infinicore/dtype.hpp

@@ -0,0 +1,22 @@
+#ifndef __INFINICORE_DTYPE_API_HPP__
+#define __INFINICORE_DTYPE_API_HPP__
+
+#include <infinicore.h>
+
+namespace infinicore {
+
+enum class DataType {
+    bfloat16 = INFINI_DTYPE_BF16,
+    float16 = INFINI_DTYPE_F16,
+    float32 = INFINI_DTYPE_F32,
+    float64 = INFINI_DTYPE_F64,
+    int32 = INFINI_DTYPE_I32,
+    int64 = INFINI_DTYPE_I64,
+    uint8 = INFINI_DTYPE_U8,
+};
+
+std::string to_string(const DataType &dtype);
+
+} // namespace infinicore
+
+#endif

include/infinicore/tensor.hpp

@@ -0,0 +1,39 @@
+#ifndef __INFINICORE_TENSOR_API_HPP__
+#define __INFINICORE_TENSOR_API_HPP__
+
+#include <vector>
+
+#include "device.hpp"
+#include "dtype.hpp"
+
+namespace infinicore {
+
+class Tensor {
+public:
+    using Size = std::size_t;
+
+    using Stride = std::ptrdiff_t;
+
+    using Shape = std::vector<Size>;
+
+    using Strides = std::vector<Stride>;
+
+    Tensor(const Shape &shape, const DataType &dtype, const Device &device);
+
+    const Shape &get_shape() const;
+
+    const DataType &get_dtype() const;
+
+    const Device &get_device() const;
+
+private:
+    Shape shape_;
+
+    DataType dtype_;
+
+    Device device_;
+};
+
+} // namespace infinicore
+
+#endif

pyproject.toml

@@ -0,0 +1,26 @@
+[build-system]
+requires = ["setuptools"]
+build-backend = "setuptools.build_meta"
+
+[project]
+name = "InfiniCore"
+version = "0.1.0"
+description = "InfiniCore 是一个跨平台统一编程工具集，为不同芯片平台的功能（包括计算、运行时、通信等）提供统一 C 语言接口。"
+readme = "README.md"
+dependencies = []
+requires-python = ">=3.10"
+classifiers = [
+    "Programming Language :: Python :: 3",
+    "License :: OSI Approved :: MIT License",
+    "Operating System :: OS Independent",
+]
+
+[project.urls]
+Homepage = "https://github.com/InfiniTensor/InfiniCore"
+Issues = "https://github.com/InfiniTensor/InfiniCore/issues"
+
+[tool.ruff]
+src = [".", "src"]
+
+[tool.ruff.lint]
+select = ["E4", "E7", "E9", "F", "I"]

setup.py

@@ -0,0 +1,30 @@
+import glob
+import os
+import subprocess
+from pathlib import Path
+
+from setuptools import setup
+from setuptools.command.build_py import build_py
+
+INSTALLATION_DIR = os.getenv("INFINI_ROOT", str(Path.home() / ".infini"))
+
+LIB_DIR = os.path.join(INSTALLATION_DIR, "lib")
+
+PACKAGE_NAME = "infinicore"
+
+PACKAGE_DIR = os.path.join(INSTALLATION_DIR, PACKAGE_NAME)
+
+
+class BuildPy(build_py):
+    def run(self):
+        subprocess.run(["xmake", "build", "-y"])
+        subprocess.run(["xmake", "install"])
+        built_lib = glob.glob(os.path.join(LIB_DIR, f"{PACKAGE_NAME}.*"))[0]
+        os.makedirs(PACKAGE_DIR, exist_ok=True)
+        self.copy_file(built_lib, PACKAGE_DIR)
+
+
+setup(
+    cmdclass={"build_py": BuildPy},
+    package_dir={"": "."},
+)

src/infinicore/device.cc

@@ -0,0 +1,33 @@
+#include <infinicore.hpp>
+
+namespace infinicore {
+
+Device::Device(const Type &type, const Index &index) : type_{type}, index_{index} {}
+
+const Device::Type &Device::get_type() const {
+    return type_;
+}
+
+const Device::Index &Device::get_index() const {
+    return index_;
+}
+
+std::string Device::to_string() const {
+    return to_string(type_) + ":" + std::to_string(index_);
+}
+
+std::string Device::to_string(const Type &type) {
+    switch (type) {
+    case Type::cpu:
+        return "cpu";
+    case Type::cuda:
+        return "cuda";
+    case Type::meta:
+        return "meta";
+    }
+
+    // TODO: Add error handling.
+    return "";
+}
+
+} // namespace infinicore

src/infinicore/dtype.cc

@@ -0,0 +1,35 @@
+#include <infinicore.hpp>
+
+namespace infinicore {
+
+std::string to_string(const DataType &dtype) {
+    std::string str{"infinicore."};
+
+    switch (dtype) {
+    case DataType::bfloat16:
+        str += "bfloat16";
+        break;
+    case DataType::float16:
+        str += "float16";
+        break;
+    case DataType::float32:
+        str += "float32";
+        break;
+    case DataType::float64:
+        str += "float64";
+        break;
+    case DataType::int32:
+        str += "int32";
+        break;
+    case DataType::int64:
+        str += "int64";
+        break;
+    case DataType::uint8:
+        str += "uint8";
+        break;
+    }
+
+    return str;
+}
+
+} // namespace infinicore