PyTorch में प्रोफाइलिंग (भाग 3): प्रोफाइल ही सब कुछ है
यह पोस्ट PyTorch में अटेंशन मैकेनिज़्म की प्रोफाइलिंग करती है, जिसमें नेव अटेंशन, इन-प्लेस ऑप्टिमाइज़ेशन और स्केल्ड डॉट प्रोडक्ट अटेंशन (SDPA) बैकएंड की तुलना की गई है। यह बताती है कि SDPA का मैथ बैकएंड, Tensor Core के कम उपयोग, मास्क पुनर्निर्माण और सेफ सॉफ्टमैक्स ओवरहेड के कारण नेव अटेंशन से धीमा है।
PyTorch
NVIDIA
Hugging Face blog29.07 · 23:03

