Michael Goin 978aed5300

[Kernel][Attention] Separate Attention.kv_scale into k_scale and v_scale (#6081)

2024-07-16 15:31:32 -07:00

..

activation.cpp

[Kernel][CPU] Add Quick gelu to CPU (#5717)

2024-06-21 06:39:40 +00:00

attention.cpp

[Kernel][Attention] Separate Attention.kv_scale into k_scale and v_scale (#6081)

2024-07-16 15:31:32 -07:00

cache.cpp

[Kernel][Attention] Separate Attention.kv_scale into k_scale and v_scale (#6081)

2024-07-16 15:31:32 -07:00

cpu_types_vsx.hpp

Support CPU inference with VSX PowerPC ISA (#5652)

2024-06-26 21:53:04 +00:00

cpu_types_x86.hpp

Support CPU inference with VSX PowerPC ISA (#5652)

2024-06-26 21:53:04 +00:00

cpu_types.hpp

Support CPU inference with VSX PowerPC ISA (#5652)

2024-06-26 21:53:04 +00:00

layernorm.cpp

[Kernel][Misc] Use TORCH_LIBRARY instead of PYBIND11_MODULE for custom ops (#5047)

2024-06-09 16:23:30 -04:00

pos_encoding.cpp

[Kernel][Misc] Use TORCH_LIBRARY instead of PYBIND11_MODULE for custom ops (#5047)

2024-06-09 16:23:30 -04:00

torch_bindings.cpp

[Kernel][Attention] Separate Attention.kv_scale into k_scale and v_scale (#6081)

2024-07-16 15:31:32 -07:00