udf-convert-to-cudf

Official

Convert Spark UDFs to GPU RapidsUDFs

AuthorNVIDIA
Version1.0.0
Installs0

System Documentation

What problem does it solve?

This Skill helps you convert an existing Apache Spark UDF into a GPU-accelerated RapidsUDF so it can run efficiently on NVIDIA GPUs while preserving CPU compatibility.

Core Features & Use Cases

  • Create the GPU UDF Wrapper: Add the required cuDF imports, implement the RapidsUDF interface, and rename the class appropriately.
  • Implement Columnar Logic: Translate row-based UDF behavior into evaluateColumnar using cuDF ColumnVector operations.
  • Validate and Harden: Update comparison tests, verify CPU and GPU results match, and check for GPU memory leaks.
  • Use Case: Apply this when you already have a tested Spark UDF and need a production-ready GPU version for accelerated Spark jobs.

Quick Start

Use this skill to convert my tested Spark UDF into a RapidsUDF and update the comparison test for GPU execution.

Dependency Matrix

Required Modules

None required

Components

references

💻 Claude Code Installation

Recommended: Let Claude install automatically. Simply copy and paste the text below to Claude Code.

Please help me install this Skill:
Name: udf-convert-to-cudf
Download link: https://github.com/NVIDIA/cudf-spark/archive/main.zip#udf-convert-to-cudf

Please download this .zip file, extract it, and install it in the .claude/skills/ directory.
View Source Repository

Agent Skills Search Helper

Install a tiny helper to your Agent, search and equip skill from 536,000+ vetted skills library on demand.