TensorComprehensions icon indicating copy to clipboard operation
TensorComprehensions copied to clipboard

A domain specific language to express machine learning workloads.

Results 90 TensorComprehensions issues
Sort by recently updated
recently updated
newest added

https://ci.pytorch.org/jenkins/job/tensorcomp-builds/job/linux-xenial-gcc5-cuda9-cudnn7-py3-conda-build-test/682/console reports an illegal memory access, but an attempt to reproduce it in https://github.com/facebookresearch/TensorComprehensions/pull/458 doesn't seem to work.

There is some interest in being able to do nested function calls i.e. define one TC and call it from inside another one. ``` lang = """ def max() {...

feature-request
in-progress

Codegen::makeLoopIterators generates names for loop iterators, but they may clash with size parameter or tensor names.

PR https://github.com/facebookresearch/TensorComprehensions/pull/183 disables SpatialBatchNorm test. It has various issues: 1. dimension mismatching for weight 2. output doesn't match in the specific reference check this is related to changes in ATen...

In the next release, we want to improve the pybindings besides other things: - [ ] PyTorch now provides the cpp extensions that we can use to extract aten tensor...

binaries
framework

Tensor Comprehensions seem like a great way to implement binary approximation of layers as in [BinaryNet](https://arxiv.org/abs/1602.02830). However, tc does not currently support bitwise operations such as xnor (^), or (|),...

enhancement
feature-request
in-progress

Here's a bit of C++ code which I'm running, which concatenates two arrays together: ``` #include #include #include #include #include #include #include #include "tc/aten/aten_compiler.h" #include "tc/autotuner/genetic_autotuner_aten.h" #include "tc/core/cuda/cuda_tc_executor.h" #include "tc/core/flags.h"...

The current [implementation](https://github.com/facebookresearch/TensorComprehensions/blob/master/tc/autotuner/parameters.cc#L333) is not portable, let's use device queried information to filter. Also make it backends-specific and split ```parameters.cc```, not in #307 though. Related to #354