mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-06 12:20:52 +01:00
416ba5c48f
Summary: ----------- - Updates current_stream and default stream API's to take `optional[device]` argument - Adds parsing logic to replace `torch.cuda.Stream` and `torch.cuda.Event` -> `torch.classes.cuda.Stream` and `torch.classes.cuda.Event` for JIT - Merges StreamContext manager for both Eager and JIT. Pull Request resolved: https://github.com/pytorch/pytorch/pull/53902 Test Plan: ------ Run JIT tests: python test/test_jit.py -v TestCUDA Run eager tests: python test/test_cuda.py -v TestCuda Reviewed By: SplitInfinity Differential Revision: D27285996 Pulled By: nikithamalgifb fbshipit-source-id: 45d9fee9a582b5f4c82330f5f99eb88584804270 |
||
|---|---|---|
| .. | ||
| builtin_functions.cpp | ||
| builtin_functions.h | ||
| canonicalize_modified_loop.cpp | ||
| canonicalize_modified_loop.h | ||
| code_template.h | ||
| concrete_module_type.cpp | ||
| concrete_module_type.h | ||
| convert_to_ssa.cpp | ||
| convert_to_ssa.h | ||
| edit_distance.cpp | ||
| edit_distance.h | ||
| error_report.cpp | ||
| error_report.h | ||
| exit_transforms.cpp | ||
| exit_transforms.h | ||
| function_schema_parser.cpp | ||
| function_schema_parser.h | ||
| inline_loop_condition.cpp | ||
| inline_loop_condition.h | ||
| ir_emitter.cpp | ||
| ir_emitter.h | ||
| lexer.cpp | ||
| lexer.h | ||
| mini_environment.h | ||
| name_mangler.cpp | ||
| name_mangler.h | ||
| parse_string_literal.h | ||
| parser_constants.h | ||
| parser.cpp | ||
| parser.h | ||
| resolver.h | ||
| schema_matching.cpp | ||
| schema_matching.h | ||
| schema_type_parser.cpp | ||
| schema_type_parser.h | ||
| script_type_parser.cpp | ||
| script_type_parser.h | ||
| source_range.cpp | ||
| source_range.h | ||
| string_to_type.cpp | ||
| strtod.cpp | ||
| strtod.h | ||
| sugared_value.cpp | ||
| sugared_value.h | ||
| tracer.cpp | ||
| tracer.h | ||
| tree_views.h | ||
| tree.h | ||
| versioned_symbols.cpp | ||
| versioned_symbols.h | ||