57 karma · joined July 1, 2013
rocminfo:
**** Agent 2 **** Name: gfx1201 Uuid: GPU-cea119534ea1127a Marketing Name: AMD Radeon Graphics Vendor Name: AMD Feature: KERNEL_DISPATCH Profile: BASE_PROFILE Float Round Mode: NEAR Max Queue Number: 128(0x80) Queue Min Size: 64(0x40) Queue Max Size: 131072(0x20000)
[32.624s](rocm-venv) a@Shark:~/github/TheRock$ ./build/dist/rocm/bin/rocm-smi
======================================== ROCm System Management Interface ======================================== ================================================== Concise Info ================================================== Device Node IDs Temp Power Partitions SCLK MCLK Fan Perf PwrCap VRAM% GPU% (DID, GUID) (Edge) (Avg) (Mem, Compute, ID) ================================================================================================================== 0 2 0x73a5, 59113 N/A N/A N/A, N/A, 0 N/A N/A 0% unknown N/A 0% 0% 1 1 0x7550, 24524 36.0°C 2.0W N/A, N/A, 0 0Mhz 96Mhz 0% auto 245.0W 4% 0% ================================================================================================================== ============================================== End of ROCm SMI Log ===============================================
Though AMD doesn't have the same "virtual ISA" as PTX right now there are increasing levels of such abstraction available in compiled flows with MLIR / Linalg etc. Those are higher level and can be compiled / jitted in realtime to obviate the need for a low level virtual ISA.
AMD Artificial Intelligence Group (AIG) leads AMD AI strategy and drives AI roadmap across client, edge, and cloud. We build AI capabilities, including silicon, software, models, use cases, to create a vibrant AMD AI ecosystem together with everyone.
Our organization, AIG SHARK (formerly nod.ai), aims to build AI software solutions that are unified, performant, flexible, and customizable. We approach the mission with open-source and community-driven principles. We believe in the power of the community and together we can build a great stack that benefits everyone. Being a part of AMD and AIG, it also means we have a wide portfolio of hardware and customers to support for realized product excellence and business impacts.
Current Open Positions:
At any given time, we have various positions posted that you may apply to: General AI runtime: https://careers.amd.com/careers-home/jobs/36717 GPU compiler/runtime: https://careers.amd.com/careers-home/jobs/37400 GPU compiler/performance: https://careers.amd.com/careers-home/jobs/39454 GPU distribution/serving: https://careers.amd.com/careers-home/jobs/38113 CPU codegeneration (New Grad): https://careers.amd.com/careers-home/jobs/38616 NPU code generation/runtime : https://careers.amd.com/careers-home/jobs/39280 General Compiler: https://careers.amd.com/careers-home/jobs/40053 Build / Infra Ninjas
That said we (Nod.ai team) will add support for xformers soon so you can opt in for xformers anyway.
It works with Pytorch -> torch-mlir -> MLIR / IREE -> vulkan. Works on both Windows and Linux. And has a simple gradio web UI https://github.com/nod-ai/SHARK/tree/main/web but we plan to enable better UI integrations very soon.
Join us on discord https://discord.gg/RUqY2h2s9u if you have any trouble. Appreciate any / all feedback.
//part of nod.ai/shark team.
//part of nod.ai / SHARK team.
IREE --> For the awesome backend to MLIR
SHARK/nod.ai --> For adapting IREE for use on various hardware and fine tuning for target hardware.
//part of nod.ai / SHARK team
If you can build torch-mlir and SHARK from src you can use it. So hopefully soon we can make pip installable packages but for now the interfaces are in constant development so you will have to build from source.
//part of nod.ai / SHARK team.