TensorComprehensions

TensorComprehensions copied to clipboard

Writes to non-zero based tensors

1

Consider: out(i) = in(10 - i) It will infer a range for i that does not necessarily start at zero. What should this mean for the domain of out? I...

abadams

Allow using scalars for bounds inference

12

We can't use scalar inputs in the bounds inference right now. So for example: ``` LANG=""" def avgpool(float(B, C, H, W) input, float kH, float kW, float sH, float sW)...

prigoyal

Issues with EBNF (grammar)

6

Hi again. First, here is how grmmar currently looks in the documentation ``` num ::= id ::= [_a-zA-Z0-9]*[_a-zA-Z][_a-zA-Z0-9]* exp ::= num | ( '-' | '!' | ... ) exp...

arogozhnikov

[wip] Register promotion improvements

2

* If a group of references was promoted into shared memory, but it could be also promoted to registers while covering exactly the same statement instances accessing it, demote it...

ftynse

Always emit the same function signature in codegen_llvm

1

thetheodor

**UNCLEAN AND NON-REBASED WORK IN PROGRESS** Do not merge or even review.

ftynse

Variable tensor sizes support for TC

6

is there a way for us to support variable tensor sizes? right now, if the tensor size changes, we have to recompile and cache. But often in computer vision, NLP,...

prigoyal

Simple Example Walkthrough

2

Existing tests run certain scripts to test certain aspects of the TC. However, there is no simple example that starts with very simple kernels like GEMM or convolution and takes...

perdavan

Similar to Preserve3Coincident necessary for GPU, except now only one outermost coincident loop is needed for CPU

wsmoses

Support for CPU computations and add is_cuda check

2

When submitting a bug report, please include the following information (where relevant): - OS: ubuntu 16.04 - How you installed TC (docker, conda, source): conda - Python version: 3.6 -...

arogozhnikov