This website requires JavaScript.
Explore
Help
Sign In
tarry.singh
/
Radix
Watch
1
Star
0
Fork
0
You've already forked Radix
Code
Issues
Pull Requests
Actions
Packages
Projects
Releases
Wiki
Activity
b244a18621
Radix
/
pjrt
History
Tarry Singh
6d720126ac
Add PJRT custom call integration with generic zmlHostBufferCallback to copy tensors to host and invoke user callbacks. Introduce Tensor.print() method to output runtime tensor values (CUDA‑specific, uses a pre‑allocated host buffer).
2023-06-05 13:42:45 +00:00
..
BUILD.bazel
Introduce a thin abstraction layer between ZML and PJRT to manage plugin loading decisions, enable compile‑time detection of linked runtimes, and handle cases such as libtpu blocking metadata access.
2023-05-15 09:36:41 +00:00
dlfcn.h
Add initial Bazel build configuration, async runtime implementation, and core MLIR dialect definitions for ZML.
2023-01-02 14:28:25 +00:00
pjrt.zig
Add PJRT custom call integration with generic zmlHostBufferCallback to copy tensors to host and invoke user callbacks. Introduce Tensor.print() method to output runtime tensor values (CUDA‑specific, uses a pre‑allocated host buffer).
2023-06-05 13:42:45 +00:00
profiler.zig
zml: clean up dead and commented code; note that copyslice is currently broken and pending reimplementation
2023-02-08 17:13:47 +00:00