Files

 main

/

python

/

Copy path

Directory actions

More options

Directory actions

More options

Latest commit

asm89

and

pytorchmergebot

caffe2: remove support for specifically running "flaky tests" (pytorc…

Feb 13, 2024

2ae655b · Feb 13, 2024

History

History

This branch is 15449 commits behind pytorch/pytorch:main.

/

python

/

Top

Folders and files

Name		Name	Last commit message	Last commit date
parent directory ..
benchmarks		benchmarks	Drop unused imports from caffe2/python (pytorch#49980)	Jan 5, 2021
docs		docs	Change caffe2 branch links to main (pytorch#100129)	Apr 27, 2023
examples		examples	[BE]: pyupgrade Python to 3.8 - imports and object inheritance only (p…	Feb 7, 2023
fakelowp		fakelowp	Lint trailing newlines (pytorch#54737)	Mar 30, 2021
helpers		helpers	Support ArgMin in c2_pt_converter	Dec 6, 2020
ideep		ideep	[CODEMOD][caffe2] replace numpy.bool with bool (pytorch#111432)	Oct 18, 2023
layers		layers	[BE] [1/3] Rewrite super() calls in caffe2 and benchmarks (pytorch#…	Feb 11, 2023
mint		mint	[typing] suppress errors in fbcode/caffe2 - batch 2	Jul 22, 2021
mkl		mkl	Drop unused imports from caffe2/python (pytorch#49980)	Jan 5, 2021
modeling		modeling	[CODEMOD][caffe2] replace uses of np.float with np.float64 (pytorch#1…	Nov 3, 2023
models		models	[BE]: enable ruff rules PLR1722 and PLW3301 (pytorch#109461)	Sep 18, 2023
onnx		onnx	s/supress/suppress/ (pytorch#119132)	Feb 4, 2024
operator_test		operator_test	[CODEMOD][caffe2] replace uses of np.float with np.float64 (pytorch#1…	Nov 3, 2023
predictor		predictor	[BE]: Cleanup deprecated stdlib imports (UP006,UP035) (pytorch#101361)	May 15, 2023
rnn		rnn	Lint trailing newlines (pytorch#54737)	Mar 30, 2021
serialized_test		serialized_test	[caffe2] replace numpy.object with object (pytorch#111494)	Oct 19, 2023
test		test	Fix possible naming collision issue (pytorch#107743)	Sep 8, 2023
trt		trt	[BE] Get rid of six in caffe2 code (pytorch#93956)	Feb 2, 2023
CMakeLists.txt		CMakeLists.txt	Move workspace related functions to separate file (pytorch#87651)	Oct 29, 2022
__init__.py		__init__.py	Revert "[Reland2] Update NVTX to NVTX3 (pytorch#109843)"	Dec 5, 2023
_import_c_extension.py		_import_c_extension.py	Drop unused imports from caffe2/python (pytorch#49980)	Jan 5, 2021
_import_c_extension.pyi		_import_c_extension.pyi	[BE]: Apply PYI autofixes to various types (pytorch#107521)	Aug 20, 2023
allcompare_test.py		allcompare_test.py	Disallow versionless Python shebangs (pytorch#58275)	May 14, 2021
attention.py		attention.py	Remove __future__ imports for legacy Python2 supports (pytorch#45033)	Sep 24, 2020
benchmark_generator.py		benchmark_generator.py	Fix ONNX ATen fallback for non-caffe2 engines	Apr 14, 2022
binarysize.py		binarysize.py	[BE]: pyupgrade Python to 3.8 - imports and object inheritance only (p…	Feb 7, 2023
brew.py		brew.py	[pytorch] Update caffe2/python to eliminate Pyre errors (pytorch#52083)	Feb 11, 2021
brew_test.py		brew_test.py	Remove __future__ imports for legacy Python2 supports (pytorch#45033)	Sep 24, 2020
build.py		build.py	Remove __future__ imports for legacy Python2 supports (pytorch#45033)	Sep 24, 2020
cached_reader.py		cached_reader.py	[BE] [1/3] Rewrite super() calls in caffe2 and benchmarks (pytorch#…	Feb 11, 2023
caffe_translator.py		caffe_translator.py	[BE]: pyupgrade Python to 3.8 - imports and object inheritance only (p…	Feb 7, 2023
caffe_translator_test.py		caffe_translator_test.py	Fix several ResourceWarning: unclosed file (pytorch#15746)	Jan 9, 2019
checkpoint.py		checkpoint.py	[BE] [1/3] Rewrite super() calls in caffe2 and benchmarks (pytorch#…	Feb 11, 2023
checkpoint_test.py		checkpoint_test.py	[BE]: Apply pyupgrade yield from and unit test alias upgrades (pytorc…	Feb 7, 2023
clean_workspace_test.py		clean_workspace_test.py	extract out the clean workspace test to its own file (pytorch#88682)	Nov 9, 2022
cnn.py		cnn.py	[BE] [1/3] Rewrite super() calls in caffe2 and benchmarks (pytorch#…	Feb 11, 2023
context.py		context.py	[BE]: pyupgrade Python to 3.8 - imports and object inheritance only (p…	Feb 7, 2023
context.pyi		context.pyi	[BE] Enable Ruff's Flake8 PYI001 (pytorch#112823)	Nov 3, 2023
context_test.py		context_test.py	caffe2: refactor context to allow being typed (pytorch#48340)	Dec 1, 2020
control.py		control.py	[BE] Get rid of future (pytorch#92596)	Jan 19, 2023
control_ops_grad.py		control_ops_grad.py	Remove __future__ imports for legacy Python2 supports (pytorch#45033)	Sep 24, 2020
control_ops_grad_test.py		control_ops_grad_test.py	Remove __future__ imports for legacy Python2 supports (pytorch#45033)	Sep 24, 2020
control_ops_util.py		control_ops_util.py	Remove __future__ imports for legacy Python2 supports (pytorch#45033)	Sep 24, 2020
control_test.py		control_test.py	[BE] [1/3] Rewrite super() calls in caffe2 and benchmarks (pytorch#…	Feb 11, 2023
convert.py		convert.py	Lint trailing newlines (pytorch#54737)	Mar 30, 2021
convert_test.py		convert_test.py	Drop unused imports from caffe2/python (pytorch#49980)	Jan 5, 2021
convnet_benchmarks.py		convnet_benchmarks.py	Remove Apache headers from source.	Mar 27, 2018
convnet_benchmarks_test.py		convnet_benchmarks_test.py	Skip convnets benchmark in rocm CI (pytorch#17331)	Feb 21, 2019
core.py		core.py	[core][perf] Use set comprehensions in _RecreateLookupTables. (pytorc…	Feb 10, 2024
core_gradients_test.py		core_gradients_test.py	Fix some typos.	Apr 11, 2022
core_test.py		core_test.py	[CODEMOD][caffe2] replace uses of np.float with np.float64 (pytorch#1…	Nov 3, 2023
crf.py		crf.py	[BE]: pyupgrade Python to 3.8 - imports and object inheritance only (p…	Feb 7, 2023
crf_predict.py		crf_predict.py	Remove __future__ imports for legacy Python2 supports (pytorch#45033)	Sep 24, 2020
crf_viterbi_test.py		crf_viterbi_test.py	Remove __future__ imports for legacy Python2 supports (pytorch#45033)	Sep 24, 2020
data_parallel_model.py		data_parallel_model.py	[BE]: pyupgrade Python to 3.8 - imports and object inheritance only (p…	Feb 7, 2023
data_parallel_model_test.py		data_parallel_model_test.py	[Codemod][[codemod] Replace third-party mock with unittest.mock] caff…	Sep 29, 2023
data_workers.py		data_workers.py	Drop some Python 2 compatibility code (pytorch#51769)	Feb 11, 2021
data_workers_test.py		data_workers_test.py	Remove __future__ imports for legacy Python2 supports (pytorch#45033)	Sep 24, 2020
dataio.py		dataio.py	[BE] [1/3] Rewrite super() calls in caffe2 and benchmarks (pytorch#…	Feb 11, 2023
dataio_test.py		dataio_test.py	Drop unused imports from caffe2/python (pytorch#49980)	Jan 5, 2021
dataset.py		dataset.py	[BE]: pyupgrade Python to 3.8 - imports and object inheritance only (p…	Feb 7, 2023
db_file_reader.py		db_file_reader.py	[BE] [1/3] Rewrite super() calls in caffe2 and benchmarks (pytorch#…	Feb 11, 2023
db_test.py		db_test.py	Remove __future__ imports for legacy Python2 supports (pytorch#45033)	Sep 24, 2020
device_checker.py		device_checker.py	[BE]: pyupgrade Python to 3.8 - imports and object inheritance only (p…	Feb 7, 2023
dlpack.h		dlpack.h	Add DLPack support for XPU backend by mapping to kDLOneAPI in DLPack … (	Mar 30, 2023
dyndep.py		dyndep.py	Remove __future__ imports for legacy Python2 supports (pytorch#45033)	Sep 24, 2020
embedding_generation_benchmark.py		embedding_generation_benchmark.py	Remove __future__ imports for legacy Python2 supports (pytorch#45033)	Sep 24, 2020
experiment_util.py		experiment_util.py	[BE]: pyupgrade Python to 3.8 - imports and object inheritance only (p…	Feb 7, 2023
extension_loader.py		extension_loader.py	Remove __future__ imports for legacy Python2 supports (pytorch#45033)	Sep 24, 2020
fakefp16_transform_lib.py		fakefp16_transform_lib.py	Remove __future__ imports for legacy Python2 supports (pytorch#45033)	Sep 24, 2020
filler_test.py		filler_test.py	Remove __future__ imports for legacy Python2 supports (pytorch#45033)	Sep 24, 2020
functional.py		functional.py	[BE]: pyupgrade Python to 3.8 - imports and object inheritance only (p…	Feb 7, 2023
functional_test.py		functional_test.py	Remove __future__ imports for legacy Python2 supports (pytorch#45033)	Sep 24, 2020
fused_8bit_rowwise_conversion_ops_test.py		fused_8bit_rowwise_conversion_ops_test.py	Fix fused_8bit_rowwise_conversion_ops_test (pytorch#98183)	Apr 4, 2023
gradient_check_test.py		gradient_check_test.py	[pytorch] Update caffe2/python to eliminate Pyre errors (pytorch#52083)	Feb 11, 2021
gradient_checker.py		gradient_checker.py	[BE]: pyupgrade Python to 3.8 - imports and object inheritance only (p…	Feb 7, 2023
gru_cell.py		gru_cell.py	[BE] [1/3] Rewrite super() calls in caffe2 and benchmarks (pytorch#…	Feb 11, 2023
hip_test_util.py		hip_test_util.py	Remove __future__ imports for legacy Python2 supports (pytorch#45033)	Sep 24, 2020
hsm_util.py		hsm_util.py	Remove __future__ imports for legacy Python2 supports (pytorch#45033)	Sep 24, 2020
hypothesis_test.py		hypothesis_test.py	[CODEMOD][caffe2] replace numpy.bool with bool (pytorch#111432)	Oct 18, 2023
hypothesis_test_util.py		hypothesis_test_util.py	[CODEMOD][caffe2] replace numpy.bool with bool (pytorch#111432)	Oct 18, 2023
ideep_test_util.py		ideep_test_util.py	Drop unused imports from caffe2/python (pytorch#49980)	Jan 5, 2021
layer_model_helper.py		layer_model_helper.py	[CODEMOD][caffe2] replace uses of np.float with np.float64 (pytorch#1…	Nov 3, 2023
layer_model_instantiator.py		layer_model_instantiator.py	Remove __future__ imports for legacy Python2 supports (pytorch#45033)	Sep 24, 2020
layer_parameter_sharing_test.py		layer_parameter_sharing_test.py	[BE]: Apply pyupgrade yield from and unit test alias upgrades (pytorc…	Feb 7, 2023
layer_test_util.py		layer_test_util.py	[BE] [1/3] Rewrite super() calls in caffe2 and benchmarks (pytorch#…	Feb 11, 2023
layers_test.py		layers_test.py	[BE]: Apply pyupgrade yield from and unit test alias upgrades (pytorc…	Feb 7, 2023
lazy.py		lazy.py	Modify lazy_dyndep loading to trigger inside workspace. (pytorch#41687)	Jul 22, 2020
lazy_dyndep.py		lazy_dyndep.py	Remove __future__ imports for legacy Python2 supports (pytorch#45033)	Sep 24, 2020
lazy_dyndep_test.py		lazy_dyndep_test.py	Disallow versionless Python shebangs (pytorch#58275)	May 14, 2021
lengths_reducer_fused_8bit_rowwise_ops_test.py		lengths_reducer_fused_8bit_rowwise_ops_test.py	Remove __future__ imports for legacy Python2 supports (pytorch#45033)	Sep 24, 2020
lengths_reducer_rowwise_8bit_ops_test.py		lengths_reducer_rowwise_8bit_ops_test.py	Remove __future__ imports for legacy Python2 supports (pytorch#45033)	Sep 24, 2020
lstm_benchmark.py		lstm_benchmark.py	Remove __future__ imports for legacy Python2 supports (pytorch#45033)	Sep 24, 2020
memonger.py		memonger.py	[BE]: Remove useless lambdas (pytorch#113602)	Nov 14, 2023
memonger_test.py		memonger_test.py	[BE]: Apply pyupgrade yield from and unit test alias upgrades (pytorc…	Feb 7, 2023
mkl_test_util.py		mkl_test_util.py	Remove __future__ imports for legacy Python2 supports (pytorch#45033)	Sep 24, 2020
model_device_test.py		model_device_test.py	Unify gpu_support variable in python tests (pytorch#16748)	Feb 7, 2019
model_helper.py		model_helper.py	[BE]: pyupgrade Python to 3.8 - imports and object inheritance only (p…	Feb 7, 2023
model_helper_test.py		model_helper_test.py	Remove __future__ imports for legacy Python2 supports (pytorch#45033)	Sep 24, 2020
modifier_context.py		modifier_context.py	[BE]: pyupgrade Python to 3.8 - imports and object inheritance only (p…	Feb 7, 2023
mpi_python.cc		mpi_python.cc	Replace c10::guts::stuff with std::stuff (pytorch#30915)	Dec 16, 2019
muji.py		muji.py	Unify cuda and hip device types in Caffe2 python front end (pytorch#1…	Nov 29, 2018
muji_test.py		muji_test.py	Unify cuda and hip device types in Caffe2 python front end (pytorch#1…	Nov 29, 2018
net_builder.py		net_builder.py	[BE] [1/3] Rewrite super() calls in caffe2 and benchmarks (pytorch#…	Feb 11, 2023
net_builder_test.py		net_builder_test.py	Enable flake8-bugbear B020 lint (pytorch#110823)	Oct 24, 2023
net_drawer.py		net_drawer.py	[BE]: Remove useless lambdas (pytorch#113602)	Nov 14, 2023
net_printer.py		net_printer.py	Fix usages of contextmanager without finally (pytorch#96170)	Mar 8, 2023
net_printer_test.py		net_printer_test.py	Remove __future__ imports for legacy Python2 supports (pytorch#45033)	Sep 24, 2020
nomnigraph.py		nomnigraph.py	[BE]: pyupgrade Python to 3.8 - imports and object inheritance only (p…	Feb 7, 2023
nomnigraph_test.py		nomnigraph_test.py	Drop unused imports from caffe2/python (pytorch#49980)	Jan 5, 2021
nomnigraph_transformations.py		nomnigraph_transformations.py	Remove __future__ imports for legacy Python2 supports (pytorch#45033)	Sep 24, 2020
nomnigraph_transformations_test.py		nomnigraph_transformations_test.py	Remove __future__ imports for legacy Python2 supports (pytorch#45033)	Sep 24, 2020
normalizer.py		normalizer.py	[BE] [1/3] Rewrite super() calls in caffe2 and benchmarks (pytorch#…	Feb 11, 2023
normalizer_context.py		normalizer_context.py	caffe2: refactor context to allow being typed (pytorch#48340)	Dec 1, 2020
normalizer_test.py		normalizer_test.py	[BE]: Apply pyupgrade yield from and unit test alias upgrades (pytorc…	Feb 7, 2023
numa_benchmark.py		numa_benchmark.py	Remove __future__ imports for legacy Python2 supports (pytorch#45033)	Sep 24, 2020
numa_test.py		numa_test.py	Remove __future__ imports for legacy Python2 supports (pytorch#45033)	Sep 24, 2020
observer_test.py		observer_test.py	Remove __future__ imports for legacy Python2 supports (pytorch#45033)	Sep 24, 2020
operator_fp_exceptions_test.py		operator_fp_exceptions_test.py	[BE]: Apply pyupgrade yield from and unit test alias upgrades (pytorc…	Feb 7, 2023
optimizer.py		optimizer.py	[fix] Disable EMA if ema_alpha is set to None (pytorch#98992)	Apr 18, 2023
optimizer_context.py		optimizer_context.py	caffe2: refactor context to allow being typed (pytorch#48340)	Dec 1, 2020
optimizer_test.py		optimizer_test.py	[BE] [1/3] Rewrite super() calls in caffe2 and benchmarks (pytorch#…	Feb 11, 2023
optimizer_test_util.py		optimizer_test_util.py	[BE]: pyupgrade Python to 3.8 - imports and object inheritance only (p…	Feb 7, 2023
parallel_workers.py		parallel_workers.py	[BE]: pyupgrade Python to 3.8 - imports and object inheritance only (p…	Feb 7, 2023
parallel_workers_test.py		parallel_workers_test.py	[BE]: Remove useless lambdas (pytorch#113602)	Nov 14, 2023
parallelize_bmuf_distributed_test.py		parallelize_bmuf_distributed_test.py	Remove __future__ imports for legacy Python2 supports (pytorch#45033)	Sep 24, 2020
pipeline.py		pipeline.py	[BE]: pyupgrade Python to 3.8 - imports and object inheritance only (p…	Feb 7, 2023
pipeline_test.py		pipeline_test.py	[BE]: Apply pyupgrade yield from and unit test alias upgrades (pytorc…	Feb 7, 2023
predictor_constants.py		predictor_constants.py	Remove __future__ imports for legacy Python2 supports (pytorch#45033)	Sep 24, 2020
pybind_state.cc		pybind_state.cc	Fix typo under caffe2 directory (pytorch#110825)	Oct 8, 2023
pybind_state.h		pybind_state.h	[Cmake] Check that gcc-9.4 or newer is used (pytorch#112858)	Nov 6, 2023
pybind_state_dlpack.cc		pybind_state_dlpack.cc	Bump dlpack.h to latest version (pytorch#65047)	Jan 21, 2022
pybind_state_dlpack.h		pybind_state_dlpack.h	Bump dlpack.h to latest version (pytorch#65047)	Jan 21, 2022
pybind_state_gpu.cc		pybind_state_gpu.cc	[NOOP][clangformat][codemod] Enable CLANGFORMAT for caffe2/caffe2/* (p…	Nov 3, 2021
pybind_state_hip.cc		pybind_state_hip.cc	[NOOP][clangformat][codemod] Enable CLANGFORMAT for caffe2/caffe2/* (p…	Nov 3, 2021
pybind_state_ideep.cc		pybind_state_ideep.cc	Update fbcode symlinks for mkl-dnn ideep 2.5.2	Mar 4, 2022
pybind_state_int8.cc		pybind_state_int8.cc	Disable avoid-non-const-global-variables lint check (pytorch#62008)	Jul 23, 2021
pybind_state_nomni.cc		pybind_state_nomni.cc	[NOOP][clangformat][codemod] Enable CLANGFORMAT for caffe2/caffe2/* (p…	Nov 3, 2021
pybind_state_registry.cc		pybind_state_registry.cc	Disable avoid-non-const-global-variables lint check (pytorch#62008)	Jul 23, 2021
pybind_state_registry.h		pybind_state_registry.h	Move registry fully to c10 (pytorch#12077)	Sep 27, 2018
pybind_workspace.cc		pybind_workspace.cc	[caffe2] Fix pybind11 native python link error (pytorch#92325)	Jan 26, 2023
pybind_workspace.h		pybind_workspace.h	[caffe2] Fix pybind11 native python link error (pytorch#92325)	Jan 26, 2023
python_op_test.py		python_op_test.py	Remove unused six code for Python 2/3 compatibility (pytorch#48077)	Dec 23, 2020
queue_util.py		queue_util.py	Remove __future__ imports for legacy Python2 supports (pytorch#45033)	Sep 24, 2020
record_queue.py		record_queue.py	[BE] [1/3] Rewrite super() calls in caffe2 and benchmarks (pytorch#…	Feb 11, 2023
recurrent.py		recurrent.py	[BE] Get rid of future (pytorch#92596)	Jan 19, 2023
regularizer.py		regularizer.py	[BE] [1/3] Rewrite super() calls in caffe2 and benchmarks (pytorch#…	Feb 11, 2023
regularizer_context.py		regularizer_context.py	caffe2: refactor context to allow being typed (pytorch#48340)	Dec 1, 2020
regularizer_test.py		regularizer_test.py	Enable FP16 sparse regularizer	Feb 12, 2021
rnn_cell.py		rnn_cell.py	[BE] [1/3] Rewrite super() calls in caffe2 and benchmarks (pytorch#…	Feb 11, 2023
schema.py		schema.py	Enable flake8-bugbear B020 lint (pytorch#110823)	Oct 24, 2023
schema_test.py		schema_test.py	[codemod][numpy] replace np.str with str (pytorch#103931)	Jun 21, 2023
scope.py		scope.py	Remove __future__ imports for legacy Python2 supports (pytorch#45033)	Sep 24, 2020
scope_test.py		scope_test.py	[BE]: Apply pyupgrade yield from and unit test alias upgrades (pytorc…	Feb 7, 2023
session.py		session.py	[BE]: pyupgrade Python to 3.8 - imports and object inheritance only (p…	Feb 7, 2023
session_test.py		session_test.py	Remove __future__ imports for legacy Python2 supports (pytorch#45033)	Sep 24, 2020
sparse_to_dense_mask_test.py		sparse_to_dense_mask_test.py	[CODEMOD][caffe2] replace uses of np.float with np.float64 (pytorch#1…	Nov 3, 2023
sparse_to_dense_test.py		sparse_to_dense_test.py	Remove __future__ imports for legacy Python2 supports (pytorch#45033)	Sep 24, 2020
task.py		task.py	[BE] [1/3] Rewrite super() calls in caffe2 and benchmarks (pytorch#…	Feb 11, 2023
task_test.py		task_test.py	[RFC][caffe2] TaskGroup.__repr__ shouldn't have side effects	Oct 1, 2020
test_util.py		test_util.py	caffe2: remove support for specifically running "flaky tests" (pytorc…	Feb 13, 2024
text_file_reader.py		text_file_reader.py	Remove __future__ imports for legacy Python2 supports (pytorch#45033)	Sep 24, 2020
timeout_guard.py		timeout_guard.py	[BE] Get rid of future (pytorch#92596)	Jan 19, 2023
toy_regression_test.py		toy_regression_test.py	Enable junk fill for the default CPU allocator (pytorch#13377)	Nov 8, 2018
transformations.py		transformations.py	[BE]: pyupgrade Python to 3.8 - imports and object inheritance only (p…	Feb 7, 2023
transformations_test.py		transformations_test.py	[BE]: Apply pyupgrade yield from and unit test alias upgrades (pytorc…	Feb 7, 2023
tt_core.py		tt_core.py	Remove __future__ imports for legacy Python2 supports (pytorch#45033)	Sep 24, 2020
tt_core_test.py		tt_core_test.py	[BE]: Apply pyupgrade yield from and unit test alias upgrades (pytorc…	Feb 7, 2023
utils.py		utils.py	[codemod][numpy] replace np.str with str (pytorch#103931)	Jun 21, 2023
utils_test.py		utils_test.py	Remove __future__ imports for legacy Python2 supports (pytorch#45033)	Sep 24, 2020
visualize.py		visualize.py	[BE]: pyupgrade Python to 3.8 - imports and object inheritance only (p…	Feb 7, 2023
workspace.py		workspace.py	[caffe2] replace numpy.object with object (pytorch#111494)	Oct 19, 2023
workspace_test.py		workspace_test.py	[caffe2] replace numpy.object with object (pytorch#111494)	Oct 19, 2023
View all files