Initial commit

Author: dmitrii-zagornyi

LICENSE.txt

@@ -0,0 +1,24 @@
+Copyright (c) 2018, Intel Corporation
+
+Redistribution and use in source and binary forms, with or without
+modification, are permitted provided that the following conditions are met:
+
+    * Redistributions of source code must retain the above copyright notice,
+      this list of conditions and the following disclaimer.
+    * Redistributions in binary form must reproduce the above copyright
+      notice, this list of conditions and the following disclaimer in the
+      documentation and/or other materials provided with the distribution.
+    * Neither the name of Intel Corporation nor the names of its contributors
+      may be used to endorse or promote products derived from this software
+      without specific prior written permission.
+
+THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
+AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
+IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
+DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE
+FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
+DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
+SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
+CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
+OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
+OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.

mkl-service/__init__.py

@@ -0,0 +1,2 @@
+from ._py_mkl_service import *
+#__all__ = ['tests']

mkl-service/_mkl_service.pxd

@@ -0,0 +1,125 @@
+ctypedef long long MKL_INT64
+ctypedef unsigned long long MKL_UINT64
+ctypedef int MKL_INT
+
+
+cdef extern from "mkl.h":
+    # MKL_INT64 mkl_peak_mem_usage(int mode)
+    # In
+    int MKL_PEAK_MEM_ENABLE
+    int MKL_PEAK_MEM_DISABLE
+    int MKL_PEAK_MEM
+    int MKL_PEAK_MEM_RESET
+
+    # int mkl_set_memory_limit(int mem_type, size_t limit)
+    # In
+    int MKL_MEM_MCDRAM
+
+    # int mkl_cbwr_set(int settings)
+    # In
+    int MKL_CBWR_AUTO
+    int MKL_CBWR_COMPATIBLE
+    int MKL_CBWR_SSE2
+    int MKL_CBWR_SSE3
+    int MKL_CBWR_SSSE3
+    int MKL_CBWR_SSE4_1
+    int MKL_CBWR_SSE4_2
+    int MKL_CBWR_AVX
+    int MKL_CBWR_AVX2
+    int MKL_CBWR_AVX512_MIC
+    int MKL_CBWR_AVX512
+    # Out
+    int MKL_CBWR_SUCCESS
+    int MKL_CBWR_ERR_INVALID_INPUT
+    int MKL_CBWR_ERR_UNSUPPORTED_BRANCH
+    int MKL_CBWR_ERR_MODE_CHANGE_FAILURE
+
+    # int mkl_enable_instructions(int isa)
+    # In
+    int MKL_ENABLE_AVX512
+    int MKL_ENABLE_AVX512_MIC
+    int MKL_ENABLE_AVX2
+    int MKL_ENABLE_AVX
+    int MKL_ENABLE_SSE4_2
+
+    # unsigned int vmlSetMode(unsigned int mode)
+    # In
+    int VML_HA
+    int VML_LA
+    int VML_EP
+    int VML_FTZDAZ_ON
+    int VML_FTZDAZ_OFF
+    int VML_ERRMODE_IGNORE
+    int VML_ERRMODE_ERRNO
+    int VML_ERRMODE_STDERR
+    int VML_ERRMODE_EXCEPT
+    int VML_ERRMODE_CALLBACK
+    int VML_ERRMODE_DEFAULT
+
+    # int vmlSetErrStatus(const MKL_INT status)
+    # In
+    int VML_STATUS_OK
+    int VML_STATUS_ACCURACYWARNING
+    int VML_STATUS_BADSIZE
+    int VML_STATUS_BADMEM
+    int VML_STATUS_ERRDOM
+    int VML_STATUS_SING
+    int VML_STATUS_OVERFLOW
+    int VML_STATUS_UNDERFLOW
+
+
+    ctypedef struct MKLVersion:
+        int MajorVersion
+        int MinorVersion
+        int UpdateVersion
+        char* ProductStatus
+        char* Build
+        char* Processor
+        char* Platform
+
+    # MKL support functions
+    void mkl_get_version(MKLVersion *pv)
+    void mkl_get_version_string(char *buf, int len)
+
+    # Threading
+    void mkl_set_num_threads(int nth)
+    int mkl_domain_set_num_threads(int nt, int domain)
+    int mkl_set_num_threads_local(int nth)
+    void mkl_set_dynamic(int flag)
+    int mkl_get_max_threads()
+    int mkl_domain_get_max_threads(int domain)
+    int mkl_get_dynamic()
+
+    # Timing
+    float second()
+    double dsecnd()
+    void mkl_get_cpu_clocks(MKL_UINT64 *clocks)
+    double mkl_get_cpu_frequency()
+    double mkl_get_max_cpu_frequency()
+    double mkl_get_clocks_frequency()
+
+    # Memory
+    void mkl_free_buffers()
+    void mkl_thread_free_buffers()
+    int mkl_disable_fast_mm()
+    MKL_INT64 mkl_mem_stat(int *buf)
+    MKL_INT64 mkl_peak_mem_usage(int mode)
+    int mkl_set_memory_limit(int mem_type, size_t limit)
+
+    # Conditional Numerical Reproducibility
+    int mkl_cbwr_set(int settings)
+    int mkl_cbwr_get_auto_branch()
+
+    # Miscellaneous
+    int mkl_enable_instructions(int isa)
+    int mkl_set_env_mode(int mode)
+    int mkl_verbose(int enable)
+    int mkl_set_mpi (int vendor, const char *custom_library_name)
+
+    # VM Service Functions
+    unsigned int vmlSetMode(unsigned int mode)
+    unsigned int vmlGetMode()
+    int vmlSetErrStatus(const MKL_INT status)
+    int vmlGetErrStatus()
+    int vmlClearErrStatus()
+