NVIDIA-Merlin / NVIDIA-Merlin/Transformers4Rec

[Task] Create unit tests for data_loader_engine='pyarrow', and no_cuda =False

Open
#286 0 comments 0 reactions 0 assignees View on GitHub

Nobody has claimed this yet.

area/pytorch Tests / CI / CD
Dominant language
Python
Stars
1.3k
Forks
165
Avg merge
1m
Merged PRs (30d)
2

Description

Description

Currently all our unit tests for data_loader_engine='pyarrow' checks the training and eval_loop on CPU by setting no_cuda=True. We need to check the same on GPU with no_cuda=False.

Contributor guide

Open the contributing guide

First steps

  1. Read the whole issue, then the project's contributing guide.
  2. Comment on the issue to say you are picking it up — it saves two people doing the same work.
  3. Fork the repository and make your change on a branch.
  4. Open a pull request that references the issue number.

Research direction

Locate the existing unit tests for data_loader_engine='pyarrow' that exercise training and eval_loop with no_cuda=True. Add equivalent GPU coverage using no_cuda=False, then run the relevant unit-test file or test selection and confirm both workflows pass with CUDA enabled.

Written by the indexing model from the issue text.

Assessment

Tech stack
python, pytorch
Domain
machine-learning, testing-qa
Issue type
Feature
Difficulty
3/5
Estimated time
1-2 days
Activity status
Stale
Clarity
Mostly clear
Newbie friendliness
35/100

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.