|
pytorch
1.8.2
About: PyTorch provides Tensor computation (like NumPy) with strong GPU acceleration and Deep Neural Networks (in Python) built on a tape-based autograd system. LTS (Long Term Support) release.
 Fossies Dox: pytorch-1.8.2.tar.gz ("unofficial" and yet experimental doxygen-generated source code documentation)  |
Classes | |
| class | build_ext |
| class | clean |
| class | concat_license_files |
| class | install |
| class | wheel_concatenate |
Functions | |
| def | report (*args) |
| def | build_deps () |
| def | check_pydep (importname, module) |
| def | configure_extension_build () |
| def | print_box (msg) |
Variables | |
| name | |
| ext_modules | |
| cmdclass | |
| tuple | python_min_version = (3, 6, 2) |
| string | python_min_version_str = '.'.join(map(str, python_min_version)) |
| bool | VERBOSE_SCRIPT = True |
| Parameters parsed from environment. More... | |
| bool | RUN_BUILD_DEPS = True |
| bool | EMIT_BUILD_WARNING = False |
| bool | RERUN_CMAKE = False |
| bool | CMAKE_ONLY = False |
| list | filtered_args = [] |
| string | arg = 'build' |
| argv | |
| cwd = os.path.dirname(os.path.abspath(__file__)) | |
| lib_path = os.path.join(cwd, "torch", "lib") | |
| third_party_path = os.path.join(cwd, "third_party") | |
| caffe2_build_dir = os.path.join(cwd, "build") | |
| rel_site_packages = distutils.sysconfig.get_python_lib(prefix='') | |
| full_site_packages = distutils.sysconfig.get_python_lib() | |
| string | cmake_python_library |
| cmake_python_include_dir = distutils.sysconfig.get_python_inc() | |
| package_name = os.getenv('TORCH_PACKAGE_NAME', 'torch') | |
| Version, create_version_file, and package_name. More... | |
| version = get_torch_version() | |
| cmake = CMake() | |
| list | install_requires |
| Building dependent libraries. More... | |
| string | missing_pydep |
| wheel_concatenate = None | |
| string | build_update_message |
| dist = Distribution() | |
| script_name | |
| script_args | |
| ok = dist.parse_command_line() | |
| extensions | |
| packages | |
| entry_points | |
| extra_install_requires | |
| encoding | |
| long_description = f.read() | |
| int | version_range_max = max(sys.version_info[1], 8) + 1 |
| description | |
| long_description_content_type | |
| package_data | |
| url | |
| download_url | |
| author | |
| author_email | |
| python_requires | |
| classifiers | |
| license | |
| keywords | |
| def setup.build_deps | ( | ) |
Definition at line 290 of file setup.py.
References caffe2.bool, tools.build_pytorch_libs.build_caffe2(), build_deps(), check_pydep(), c10::aten.format(), report(), and c10::prim.zip().
Referenced by build_deps().
| def setup.check_pydep | ( | importname, | |
| module | |||
| ) |
Definition at line 366 of file setup.py.
References check_pydep().
Referenced by build_deps(), and check_pydep().
| def setup.configure_extension_build | ( | ) |
Configures extension build options according to system environment and user's choice. Returns: The input to parameters ext_modules, cmdclass, packages, and entry_points as required in setuptools.setup.
Definition at line 629 of file setup.py.
References tools.setup_helpers.env.check_env_flag(), configure_extension_build(), torch.distributed.rpc.api.defaultdict, and compare-fastrnn-results.str.
Referenced by configure_extension_build().
| def setup.print_box | ( | msg | ) |
Definition at line 796 of file setup.py.
References c10::aten.format(), c10::aten.len(), max, and at.print().
| def setup.report | ( | * | args | ) |
Definition at line 247 of file setup.py.
References at.print(), and report().
Referenced by torch::jit::SimpleValue.attr(), build_deps(), setup.build_ext.build_extensions(), setup.build_ext.get_outputs(), report(), caffe2.python.checkpoint.CheckpointManager.report_checkpoint_stats(), caffe2.python.checkpoint.MultiNodeCheckpointManager.report_checkpoint_stats(), and setup.build_ext.run().
| string setup.arg = 'build' |
Definition at line 234 of file setup.py.
Referenced by at._crash_if_asan(), torch::python.add_module_bindings(), caffe2::python.addGlobalMethods(), caffe2::python.addNomnigraphMethods(), caffe2::python.addObjectMethods(), c10::lib::cpp17::detail_.addressof(), torch::jit::ArgumentSpec.addTensor(), caffe2.annotateDefWithReadCounts(), torch.append_overloaded_arg(), at::IterArgs< F >.apply(), caffe2::memonger::ComputeBlobRecyclingForDag.apply_asyncif_blob_assignments(), caffe2::memonger::ComputeBlobRecyclingForDag.apply_recurrent_blob_assignments(), at.areAllReturnsTensors(), at.areAnyArgumentsTensorList(), torch::jit::tensorexpr::BlockAnalysis.areBufsInMap(), caffe2::ArgumentHelper.ArgumentHelper(), c10::detail_::impl< Ts >.assign(), c10::detail_::assignment< Traits >.assign_alt(), caffe2::OpenCLContext.BuildArgumentList(), C10_DEFINE_bool(), at::autocast.cached_cast(), c10::impl::detail::make_optional_tensor_explicit< at::Tensor >.call(), c10::impl::detail::make_optional_tensor_explicit< KernelType >.call(), torch::jit.canonicalSchemaString(), caffe2::RedisStoreHandler.check(), at::native::needs_dynamic_casting< func_t, nargs >.check(), at::native.check_1d(), torch::jit.checkMutableFunctionDefault(), torch::jit.checkONNXCompatibility(), caffe2::onnx::Caffe2Backend.CheckOpSchemaArguments(), c10::FunctionSchema.checkSchema(), c10::FunctionSchema.cloneWithRemappedTypes(), torch::jit::ArgumentSpec.combineHash(), caffe2.convertToCaffe2Proto(), caffe2::onnx::OnnxExporter.CopyCaffe2ArgToOnnxAttr(), torch::jit::ArgumentSpecCreator.create(), torch::jit::Module.create_class(), caffe2::onnx::OnnxExporter.CreateBinaryElementwiseOpNodes(), caffe2::onnx::OnnxExporter.CreateCastNodes(), caffe2::onnx::OnnxExporter.CreateChannelShuffleNodes(), caffe2::onnx::OnnxExporter.CreateConcatNodes(), caffe2::onnx::Caffe2Backend.CreateDynamicSlice(), caffe2::onnx::Caffe2Backend.CreateLRN(), nom::Graph< T, U >.createNode(), caffe2.CreateOperatorDef(), caffe2::onnx::Caffe2Backend.CreateSlice(), torch::jit.createStackForSchema(), at::TensorIteratorBase.data_ptr(), caffe2::emulator::DataRandomFiller.DataRandomFiller(), torch::class_< CurClass >.def(), torch::class_< CurClass >.def_pickle(), at.deleter(), at::TensorIteratorBase.device(), at::TensorIteratorBase.device_type(), torch::PythonArgs.dimnamelist(), at::TensorIteratorBase.dtype(), caffe2::ProfileOperatorObserver.Dump(), caffe2::opt.DumpGraph(), at::TensorIteratorBase.element_size(), elementwiseAnd(), elementwiseEQ(), elementwiseOr(), elementwiseXor(), torch::jit::to_ir.emitApplySpecialForm(), torch::jit::to_ir.emitFormalArguments(), ska::flat_hash_set< T, H, E, A >.emplace(), ska_ordered::flat_hash_set< T, H, E, A >.emplace(), caffe2::ExecutionOptions.ExecutionOptions(), torch::jit::fuser.expandArgs(), expired(), caffe2::python::python_detail.feedBlob(), torch::jit.findArgument(), at::autocast.firstarg_is_eligible(), torch.format_invalid_args(), freeWeakRef(), caffe2.freezeQuantizationParams(), caffe2::opt.fuseNNPACKConvRelu(), at::TensorIteratorBase.get_data_ptrs(), at::TensorIteratorBase.get_strides(), caffe2.GetArgumentIndex(), caffe2::Converter.getArgumentsFromOperator(), caffe2::cast.GetCastDataType(), torch::PythonArgs.getDoublelist(), caffe2.GetFlagArgument(), caffe2.GetGradientForOp(), caffe2.getLayout(), torch::jit::fuser.getMapSize(), caffe2::BackendTransformerBase.getModelId(), caffe2.GetMutableArgumentImpl(), torch::jit.getSchemaWithNameAndDefaults(), caffe2::HandleExecutorThreadExceptionsGuard.globalInit(), torch.handle_torch_function(), torch.handle_torch_function_no_python_arg_parser(), c10::FunctionSchema.hasAnyAliasInfo(), torch::cuda::python.initCommMethods(), torch::impl::dispatch.initDispatchBindings(), torch::jit::tensorexpr::BlockCodeGen.Initialize(), torch::jit.initJitBackendBindings(), torch::jit.initJITBindings(), torch::jit.initJitScriptBindings(), caffe2::onnx::OnnxExporter.InitOpToTensorProto(), torch::jit.initPythonIRBindings(), torch::jit::tracer.initPythonTracerBindings(), at::TensorIteratorBase.input(), at::TensorIteratorBase.input_dtype(), torch::PythonArgs.intlistWithDefault(), at::TensorIteratorBase.is_cpu_scalar(), at::autocast.is_eligible(), c10::FunctionSchema.is_mutable(), at::TensorIteratorBase.is_scalar(), caffe2::onnx::OnnxExporter.IsBlockListed(), torch::jit.isIntOrFloatUsedAsList(), torch::utils.legacy_tensor_ctor(), torch::utils.legacy_tensor_new(), torch::PythonArgs.list_of_optional_tensors(), quant_utils.MakeArgForConv1d(), caffe2.MakeArgument(), at::TensorIteratorBase.mark_outputs(), torch::jit::tensorexpr::IRMutator.mutate(), newWithWeakPtr(), caffe2::onnx::OnnxAttributes.OnnxAttrToCaffe2Arg(), caffe2::RecurrentNetworkExecutorBase.op_deps(), at::IterArgs< F >.operator()(), c10::guts::detail::_identity.operator()(), c10.operator<<(), c10::variant< Ts >.operator=(), torch::nn::ModuleHolder< Contained >.operator[](), at::TensorIteratorBase.output(), torch::jit::ScriptTypeParser.parseArgsFromDecl(), c10.ParseCommandLineFlags(), torch.parseDimnameList(), torch::jit.percentFormat(), torch::jit::tensorexpr::TensorExprKernel.prepareBufferArgs(), torch::jit.preprocessCaffe2Ops(), torch::jit::tensorexpr::BlockPrinter.PrintArguments(), torch::jit::PythonPrintImpl.printDefaultValue(), torch::jit::PythonPrintImpl.printFunction(), torch::jit::PythonPrintImpl.printNamedType(), torch::jit::PythonPrintImpl.printRHS(), at::native::detail::MinMaxReductionOps< comp_t >.project(), at::native::detail::ArgReductionOps< comp_t >.project(), torch::jit::fuser::cuda::KernelArgumentHolder.push(), torch::jit.push_one(), PYBIND11_MODULE(), caffe2.PYBIND11_MODULE(), torch::utils.recursive_apply(), at::native::detail::MinMaxReductionOps< comp_t >.reduce(), at::TensorIteratorBase.remove_operand(), at::TensorIteratorBase.reorder_dimensions(), caffe2::onnx.rewriteSubnet(), torch::PythonArgs.scalar_slow(), at::TensorIteratorBase.scalar_value(), torch::PythonArgs.scalarlist(), cl::detail::SetArg< index, T0 >.set(), torch::autograd.set_anomaly_mode_enabled(), torch::autograd.set_autocast_enabled(), torch::autograd.set_forward_AD_enabled(), torch::autograd.set_grad_enabled(), at::autocast.set_opt_dtype(), caffe2.setSpecialArgs(), torch::jit::to_ir.shouldDeriveSetStateType(), c10.simpleClassTypeArg(), torch::jit::ArgumentSpecCreator.specializeTypes(), caffe2.splitSizes(), caffe2::PerfNetObserver.Stop(), at::TensorIteratorBase.strides(), c10::guts.tail(), c10::guts::detail.tail_(), at::TensorIteratorBase.tensor(), torch::tensors.Tensor_instancecheck(), torch::PythonArgs.tensorlist(), torch::PythonArgs.tensorlist_n(), caffe2.TEST(), THCPModule_hasPrimaryContext(), THCPModule_memoryStats(), THCPModule_resetAccumulatedMemoryStats(), THCPModule_resetPeakMemoryStats(), THCPModule_setDevice_wrap(), THPAutograd_initExtension(), THPFunction_apply(), THPModule_crashIfATenASAN(), THPModule_crashIfCsrcASAN(), THPModule_crashIfCsrcUBSAN(), THPModule_has_torch_function(), THPModule_initNames(), THPModule_set_display_vmap_fallback_warnings_mode(), THPModule_setAllowTF32CuBLAS(), THPModule_setAllowTF32CuDNN(), THPModule_setBackcompatBroadcastWarn(), THPModule_setBackcompatKeepdimWarn(), THPModule_setBenchmarkCuDNN(), THPModule_setDeterministicAlgorithms(), THPModule_setDeterministicCuDNN(), THPModule_setFlushDenormal(), THPModule_setNumInteropThreads(), THPModule_setNumThreads(), THPModule_setQEngine(), THPModule_setUserEnabledCuDNN(), THPModule_setUserEnabledMkldnn(), THPUtils_checkIntTuple(), THPUtils_getCallable(), THPUtils_tryUnpackLongs(), THPUtils_tryUnpackLongVarArgs(), THPUtils_unpackIntTuple(), THPUtils_unpackLongs(), THPUtils_unpackSize(), torch::autograd.THPVariable_apply_(), torch::autograd.THPVariable_from_numpy(), caffe2::Worker.ThreadFunc(), torch::jit.toPyObject(), caffe2::TvmTransformer.transform(), torch::jit.tryMatchArgument(), torch::jit.tryMatchSchema(), at::autocast.type_from_firstarg(), unpack_input(), at::TensorIteratorBase.unsafe_replace_operand(), torch::jit::fuser::cuda::executor_utils.validateKernelInputs(), torch::jit::fuser::cuda::executor_utils.validateKernelOutputs(), torch::jit.varargsCanBeUsedAsList(), caffe2::OpSchema.Verify(), torch::jit::tensorexpr::IRVisitor.visit(), and torch::jit::tensorexpr::IRPrinter.visit().
| string setup.build_update_message |
| setup.cmake_python_include_dir = distutils.sysconfig.get_python_inc() |
| string setup.cmake_python_library |
| setup.dist = Distribution() |
Definition at line 807 of file setup.py.
Referenced by torch::throughput_benchmark::detail::BenchmarkHelper< Input, Output, Model >.benchmark(), torch::jit::Node.blocksFromGraphBlock(), caffe2::TensorFiller.Dist(), caffe2::DropoutOp< T, Context >.RunOnDevice(), caffe2::MarginRankingCriterionGradientOp< Context >.RunOnDevice(), caffe2::SparseDropoutWithReplacementOp< Context >.RunOnDevice(), and TEST().
| setup.full_site_packages = distutils.sysconfig.get_python_lib() |
| setup.install_requires |
| string setup.missing_pydep |
| setup.ok = dist.parse_command_line() |
Definition at line 811 of file setup.py.
Referenced by c10d::ProcessGroupMPI::AsyncWork.wait().
| setup.package_name = os.getenv('TORCH_PACKAGE_NAME', 'torch') |
| string setup.python_min_version_str = '.'.join(map(str, python_min_version)) |
| setup.rel_site_packages = distutils.sysconfig.get_python_lib(prefix='') |
| bool setup.VERBOSE_SCRIPT = True |
| int setup.version_range_max = max(sys.version_info[1], 8) + 1 |
| setup.wheel_concatenate = None |
1.9.2