pytorch/test/cpp
Mikhail Zolotukhin 35e7efeb9a [TensorExpr] Add CUDA codegen. (#34227)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/34227

This PR adds a CUDA support to tensor expressions.

Differential Revision: D20251836

Test Plan: Imported from OSS

Pulled By: ZolotukhinM

fbshipit-source-id: ab36a55834cceff30c8371fef6cca1054a32f017
2020-03-16 11:49:29 -07:00
..
api [C++ API Parity] [Optimizers] added closure to optimizers (#34790) 2020-03-16 07:51:44 -07:00
common
dist_autograd [Dist Autograd] Functional API for Dist Autograd and Dist Optimizer (#33711) 2020-02-26 19:08:28 -08:00
jit Enable JIT tests on Windows (#27029) 2020-03-16 11:26:21 -07:00
rpc [pytorch-rpc] WireSerializer should check has_storage() (#34626) 2020-03-12 11:35:21 -07:00
tensorexpr [TensorExpr] Add CUDA codegen. (#34227) 2020-03-16 11:49:29 -07:00
__init__.py