TensorComprehensions

TensorComprehensions copied to clipboard

Python API

1

Is there any python api?

Zumbalamambo

resolves #230

thetheodor

[WIP] use templated isl types

8

Templated isl types require the user to specify the domain and range universes of isl objects, allowing the compiler to check whether it makes sense to combine pairs of objects....

skimo-openhub

[Cuda Codegen] Emit launch bounds

4

Cuda functions can be annotated with launch bounds, that is the maximum number of threads per block (the minimum blocks per multiprocessor can also be specified). This information is used...

thetheodor

Add experimental compilation options tuners in Python

2

These commits implement more functions in the python binders, and also add different options tuning experiments in Python.

Artix18

Proceeding with ScheduleTree evolution plan (#553) This PR replaces overloaded comparison operators with two functions, `nodeEquals` and `treeEquals`. The main motivation for this change: after grafting ScheduleTreeElem* onto ScheduleTree in...

ftynse

I got build errors while building CXX object tc/core/CMakeFiles/tc_core_cpu.dir/polyhedral/llvm_jit.cc.o. My settings are in below: - OS: ubuntu 18.04 - How you installed TC (docker, conda, source): source - Python version:...

insuhan

[DO NOT MERGE] attempt at reproducing error

3

skimo-openhub

Add UpSample example

3

@nicolasvasilache This is the C++ test for UpSample operator. In OSS env, the test is passing for me without the runtime error. Can you try it out and let me...

mingzhe09088

Add support for strided tensors

11

This commit is to start support for strided tensors. I made changes to percolate a vector in TensorInfo down to emitCudaKernel to allow codegen to cast strided tensors. This required...

protonu