Open deep learning compiler stack for cpu, gpu and specialized accelerators
javascript
machine-learning
performance
deep-learning
metal
compiler
gpu
vulkan
opencl
tensor
spirv
rocm
tvm
-
Updated
Apr 30, 2021 - Python
In numba/stencils/stencil.py, there are various places (like line 552, "if isinstance(kernel_size[i][0], int):") where we check for "int" in relation to neighborhoods. I ran across a case where I was creating a neighborhood tuple by extracting values from a Numpy array. This causes a problem because those Numpy values will not match in these isinstance int checks. I worked around it by conver