TensorComprehensions

TensorComprehensions copied to clipboard

Support for "sumExp(x) +=! exp( ... )"

12

Hi everyone, First of all, let me thank you for putting so much effort into this accessible, versatile library. As a mathematician working at the intersection between shape analysis and...

jeanfeydy

MPI suppot

4

I apologize if this question is not relevant to this forum. Does this framework support MPI? Thanks,

abidmalikwaterloo

Tensor Comprehensions Github Issues Guidelines ---------------------------------------------- If you have a feature request or a bug report (build issue), please open an issue on Github and fill the template below so...

seongwook-ham

#138 only adds a skeleton for the CPU test and smoke checks compilation. Make this into a real test.

nicolasvasilache

Padding/Boundary conditions

2

We need some way to express padding of input tensors. The current thought is a primitive pad(expr, expr), which returns the first argument *if it does no out-of-bounds reads*, and...

abadams

Support for triangular dependencies in TC language

1

one of our users reported on slack channel that they were trying to translate the following Numpy code ``` def A_matvec_batch(A, X): n, m = X.shape Y = np.zeros((n, m))...

prigoyal

Support for RNN loops

1

There is a big interest in having support for RNN loops in TC. Creating this master task to discuss further and track progress

prigoyal

Unable to use same variable for reduction twice or more in TC

2

do we not allow people to use same variable twice? also repro is available in #61 ``` FAILS: def softmax(float(N, D) I) -> (O, tmp) { tmp(n) max=! I(n, d)...

prigoyal

Handle temporaries

1

right now, everything is either an input or output. Reason being TC does not do any allocation by itself. We should find a better way to handle this cc Albert...

prigoyal

Scattering

2

We currently support computed expressions in the indices on the RHS. We should plumb through computed expressions on the LHS too, e.g: hist(im(i)) +=! 1

abadams