pytorch

Форк

AJ
Animesh Jain
[dynamo][compile-time] Remove unnecessary tree_map_only (#121052)
год назад
b7f252

Описание

Tensors and Dynamic neural networks in Python with strong GPU acceleration

Языки

Python

C
PowerShell
Cuda
PureBasic
Starlark
Dockerfile
Metal
Vim Script
JavaScript
C++
Objective-C
Assembly
GLSL
Jupyter Notebook
Makefile
Objective-C++
Smarty
LLVM
CSS
HTML
Shell
Batchfile
CMake
Java
Ruby

[ONNX][dynamo_export] Skip instance_norm decomp for export (#120866)

год назад

.circleci

Move TRITON_CONSTRAINT setting to common binary_populate_env.sh, BE - Cleanup unused build scripts (#120744)

год назад

.ctags.d

.devcontainer

[Dev Container]Add readme for devcontainer (#108848)

2 года назад

.github

TD outside of test job (#118250)

год назад

.vscode

[3/3] Update `.pyi` Python stub files and enable `'UFMT'` linter (#95268)

2 года назад

android

Add some tests for 3d channels last (#118283)

год назад

aten

[Quant][Inductor] Enable lowering of dynamic qlinear for X86Inductor (#120605)

год назад

benchmarks

Revert "[PT2D] Make the speedup benchmark works with DDP + CompiledAutograd (#120454)"

год назад

binaries

Remaining replacement of c10::stoi with std::stoi (#109482)

2 года назад

c10

add int4 packed gemm support on CPU device (#117475)

год назад

caffe2

[rocm][cmake] retrieve rocm location from ROCM_SOURCE_DIR env if specified (#120898)

год назад

cmake

[rocm][cmake] correctly check the ROCM_SOURCE_DIR environment (#120858)

год назад

docs

[dtensor] add support for loss parallel (#119877)

год назад

functorch

[BE][Ez] Update ruff to 0.2.2 (#120517)

год назад

ios

Add some tests for 3d channels last (#118283)

год назад

modules

Fix ouput typos (#120870)

год назад

mypy_plugins

More progress on type checking ValueRanges (#118870)

год назад

scripts

[easy] Reword unexpected success error messages and generated github issues now that we have sentinel files (#120766)

год назад

test

[DCP] deletes legacy formatting test (#120127)

год назад

third_party

[cuDNN] bump cuDNN-frontend submodule to 1.1.2 (#120761)

год назад

tools

TD outside of test job (#118250)

год назад

torch

[dynamo][compile-time] Remove unnecessary tree_map_only (#121052)

год назад

torchgen

[executorch] Add support for method variants in aten executorch code gen (#121016)

год назад

.bazelignore

put third_party/ittapi/ in .bazelignore (#101364)

2 года назад

.bazelrc

[bazel] add python targets (#101003)

2 года назад

.bazelversion

update Bazel to the latest release 6.1.1 (#96955)

2 года назад

.buckconfig.oss

[BE] Add regression test for aten shared build (#96697)

2 года назад

.clang-format

[BE][MPS] Add MPS to clang format (#96562)

2 года назад

.clang-tidy

Enable nested namespace check in clang-tidy (#118506)

год назад

.cmakelintrc

.coveragerc

.flake8

Update to TorchFix 0.4.0 (#119424)

год назад

.gdbinit

.git-blame-ignore-revs

Add 116583 to `.git-blame-ignore-revs` (#116676)

год назад

.gitattributes

.gitignore

[MPS] fix float32 error on mps, in linalg.matrix_rank and linalg.pinv (#114771)

год назад

.gitmodules

Revert "[Reland2] Update NVTX to NVTX3 (#109843)"

год назад

.isort.cfg

.lintrunner.toml

[BE][Ez]: Update ruff to 0.3.0 (#121003)

год назад

.lldbinit

Add helpful pretty pretting summaries to torch for lldb debugging (#97101)

2 года назад

BUCK.oss

BUILD.bazel

[ROCm] TunableOp (#114894)

год назад

CITATION.cff

CMakeLists.txt

[EZ] Fix typo in gcc version detection (#120489)

год назад

CODEOWNERS

Add myself to CODEOWNERS for serialization-related files (#118892)

год назад

CODE_OF_CONDUCT.md

CONTRIBUTING.md

[DevX] Add tool and doc on partial debug builds (#116521)

год назад

Dockerfile

Dockerfile; Add cuda bin to PATH (#117105)

год назад

GLOSSARY.md

LICENSE

MANIFEST.in

fix citation file in MANIFEST (#89994)

2 года назад

Makefile

make triton uses the wheel script now (#97995)

2 года назад

NOTICE

Add uint8 support for interpolate for CPU images (#90771)

2 года назад

README.md

Fix broken link in README (#120698)

год назад

RELEASE.md

Add Python 3.12 as experimental to release 2.2 (#119705)

год назад

SECURITY.md

Update SECURITY.MD (#93313)

2 года назад

WORKSPACE

[BE] [cuDNN] Always build assuming cuDNN >= 8.1 (#95722)

год назад

aten.bzl

add explicit vectorization for Half dtype on CPU (#96076)

2 года назад

buckbuild.bzl

Reify view_func() closures as ViewFuncs (#118404)

год назад

build.bzl

Reify view_func() closures as ViewFuncs (#118404)

год назад

build_variables.bzl

add int4 packed gemm support on CPU device (#117475)

год назад

c2_defs.bzl

[caffe2] Add option for build_cpukernel_avx2 (#116008)

год назад

c2_test_defs.bzl

defs.bzl

Remove unused build system checks and definitions (#109711)

2 года назад

docker.Makefile

docker: Only match tags that start with v* (#120670)

год назад

mypy-strict.ini

Use strict to toggle strict options in MYPYSTRICT (#118479)

год назад

mypy.ini

More progress on type checking ValueRanges (#118870)

год назад

pt_ops.bzl

[xplat][buck2][typing] Fix typechecker issue (#108525)

2 года назад

pt_template_srcs.bzl

Reify view_func() closures as ViewFuncs (#118404)

год назад

pyproject.toml

[BE]: Enable ruff LOG checks (#120674)

год назад

pytest.ini

Reduce pytest prints (#117069)

год назад

requirements-flake8.txt

Update to TorchFix 0.4.0 (#119424)

год назад

requirements.txt

[Dynamo] Analyze triton kernels via tracing to determine mutations (#117300)

год назад

setup.py

[Torchelastic][Logging] Pluggable logsspecs using python entrypoints and option to specify one by name. (#120942)

год назад

ubsan.supp

Upgrade Pybind submodule to 2.10.4 (#103989)

2 года назад

ufunc_defs.bzl

version.txt

[releng] version 2.2 -> 2.3 (#115446)

год назад

README.md

Описание

Tensors and Dynamic neural networks in Python with strong GPU acceleration

Языки

Python

C
PowerShell
Cuda
PureBasic
Starlark
Dockerfile
Metal
Vim Script
JavaScript
C++
Objective-C
Assembly
GLSL
Jupyter Notebook
Makefile
Objective-C++
Smarty
LLVM
CSS
HTML
Shell
Batchfile
CMake
Java
Ruby

Сообщить о нарушении

Описание

Tensors and Dynamic neural networks in Python with strong GPU acceleration

Языки

Python

C
PowerShell
Cuda
PureBasic
Starlark
Dockerfile
Metal
Vim Script
JavaScript
C++
Objective-C
Assembly
GLSL
Jupyter Notebook
Makefile
Objective-C++
Smarty
LLVM
CSS
HTML
Shell
Batchfile
CMake
Java
Ruby

Сообщить о нарушении

pytorch

AJAnimesh Jain[dynamo][compile-time] Remove unnecessary tree_map_only (#121052)год назадb7f252

Описание

Языки

Описание

Языки

Описание

Языки

Использование cookies

AJ
Animesh Jain
[dynamo][compile-time] Remove unnecessary tree_map_only (#121052)
год назад
b7f252