pytorch

mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 21:14:14 +08:00

Files

Mu-Chu Lee 2291199e9b [AOTInductor] Use CudaCachingAllocator for memory allocation (#162893 )

Summary:
Use c10::CudaCachingAllocator for AOTInductor's initial constant buffer
allocation.

Test Plan:
Activate test under test/cpp/aoti_inference/test.cpp

Reviewers:

Subscribers:

Tasks:

Tags:

Pull Request resolved: https://github.com/pytorch/pytorch/pull/162893
Approved by: https://github.com/desertfire

2025-09-17 17:08:20 +00:00

aoti_abi_check

[Reland] Migrate ScalarType to headeronly (#159911 )

2025-08-06 07:36:37 +00:00

aoti_inference

[AOTInductor] Use CudaCachingAllocator for memory allocation (#162893 )

2025-09-17 17:08:20 +00:00

api

Feature: Implement support for cudnn_batch_norm_out kernel to replace the autogen approach. (#123020 )

2025-08-04 22:40:33 +00:00

c10d

[fr] [xpu] Add FlightRecorder support for ProcessGroupXCCL (#158568 )