MCPcopy Create free account
hub / github.com/OpenDriveLab/ReSim / test_triton_rotary_embedding.py

File test_triton_rotary_embedding.py

SwissArmyTransformer/tests/test_triton_rotary_embedding.py:None–None  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

1import math
2import os
3import torch
4import torch.nn.functional as F

Callers

nothing calls this directly

Calls 10

FastRotaryEmbeddingClass · 0.90
RotaryEmbeddingClass · 0.90
apply_rotary_pos_emb_bhsFunction · 0.90
SatRotaryEmbeddingClass · 0.85
apply_rotary_pos_emb_index_bhsFunction · 0.85
deviceMethod · 0.80
toMethod · 0.80
printFunction · 0.50
maxMethod · 0.45
backwardMethod · 0.45

Tested by

no test coverage detected