pytorch/test/cpp
Raghavan Raman 70c9146c40 [nnc] Update block and thread extents in cuda_codegen to use int64_t (#71428)
Summary:
The block and thread extent calculations in `cuda_codegen` should be using `int64_t` instead of `int`. The updated test, `test_dynamic_shapes`, fails without this change.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/71428

Reviewed By: samdow

Differential Revision: D33640374

Pulled By: navahgar

fbshipit-source-id: 64c340ad2a9a1fa1fe066cf1c5dfc3b546b7be6d
(cherry picked from commit 6ea546ce11)
2022-01-19 23:21:24 +00:00
..
api Add flag to optionally average output attention weights across heads (#70055) 2022-01-06 17:32:37 -08:00
c10d Add support for deleteKey for FileStore (#69953) 2022-01-07 06:20:59 -08:00
common
dist_autograd Fix distributed autograd gradients synchronization (#57792) 2021-05-09 17:32:59 -07:00
jit empty_cpu: Add functions that don't depend on Tensor (#70613) 2022-01-19 00:01:58 +00:00
lazy [LT] Remove torch::lazy::convertShapes (#71291) 2022-01-14 12:06:39 -08:00
lite_interpreter_runtime Back out "Revert D30710710: [Pytorch Edge] Support profiling kineto events from external source" (#66421) 2021-10-12 10:55:29 -07:00
monitor torch/monitor: add pybind (#69567) 2022-01-12 13:35:11 -08:00
rpc Remove ProcessGroup from TensorPipeAgent initialization (#68128) 2021-11-11 12:28:55 -08:00
tensorexpr [nnc] Update block and thread extents in cuda_codegen to use int64_t (#71428) 2022-01-19 23:21:24 +00:00
__init__.py remediation of S205607 2020-07-17 17:19:47 -07:00