Skip to content

Adding helper function for enabling RocksDB Checkpoint #4213

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Closed
wants to merge 1 commit into from

Conversation

Raahul46
Copy link

Summary:
Design doc: https://docs.google.com/document/d/149LdAEHOLP7ei4hwVVkAFXGa4N9uLs1J7efxfBZp3dY/edit?tab=t.0#heading=h.49t3yfaqmt54

Context:
We are enabling the usage of rocksDB checkpoint feature in KVTensorWrapper. This allows us to create checkpoints of the embedding tables in SSD. Later, these checkpoints are used by the checkpointing component to create a checkpoint and upload it it to the manifold

In this diff:

We are adding all the helper functions to create folder structures in SSDs for storing the rocksDB checkpoints

Differential Revision: D75489819

Copy link

netlify bot commented May 29, 2025

Deploy Preview for pytorch-fbgemm-docs ready!

Name Link
🔨 Latest commit ec6049c
🔍 Latest deploy log https://app.netlify.com/projects/pytorch-fbgemm-docs/deploys/6841dc9f787c7f0008de13fd
😎 Deploy Preview https://deploy-preview-4213--pytorch-fbgemm-docs.netlify.app
📱 Preview on mobile
Toggle QR Code...

QR Code

Use your smartphone camera to open QR code link.

To edit notification comments on pull requests, go to your Netlify project configuration.

@facebook-github-bot
Copy link
Contributor

This pull request was exported from Phabricator. Differential Revision: D75489819

@facebook-github-bot
Copy link
Contributor

This pull request was exported from Phabricator. Differential Revision: D75489819

Raahul46 pushed a commit to Raahul46/FBGEMM-1 that referenced this pull request May 29, 2025
Summary:

X-link: facebookresearch/FBGEMM#1288

Design doc: https://docs.google.com/document/d/149LdAEHOLP7ei4hwVVkAFXGa4N9uLs1J7efxfBZp3dY/edit?tab=t.0#heading=h.49t3yfaqmt54

Context:
We are enabling the usage of rocksDB checkpoint feature in KVTensorWrapper. This allows us to create checkpoints of the embedding tables in SSD. Later, these checkpoints are used by the checkpointing component to create a checkpoint and upload it it to the manifold 

In this diff:

We are adding all the helper functions to create folder structures in SSDs for storing the rocksDB checkpoints

Differential Revision: D75489819
@Raahul46 Raahul46 force-pushed the export-D75489819 branch from e83a901 to bcd7ccd Compare May 29, 2025 05:30
@facebook-github-bot
Copy link
Contributor

This pull request was exported from Phabricator. Differential Revision: D75489819

@Raahul46 Raahul46 force-pushed the export-D75489819 branch from bcd7ccd to d598938 Compare May 30, 2025 22:20
@facebook-github-bot
Copy link
Contributor

This pull request was exported from Phabricator. Differential Revision: D75489819

Raahul46 pushed a commit to Raahul46/FBGEMM-1 that referenced this pull request Jun 3, 2025
Summary:

X-link: facebookresearch/FBGEMM#1288

Design doc: https://docs.google.com/document/d/149LdAEHOLP7ei4hwVVkAFXGa4N9uLs1J7efxfBZp3dY/edit?tab=t.0#heading=h.49t3yfaqmt54

Context:
We are enabling the usage of rocksDB checkpoint feature in KVTensorWrapper. This allows us to create checkpoints of the embedding tables in SSD. Later, these checkpoints are used by the checkpointing component to create a checkpoint and upload it it to the manifold 

In this diff:

We are adding all the helper functions to create folder structures in SSDs for storing the rocksDB checkpoints

Reviewed By: duduyi2013

Differential Revision: D75489819
Raahul46 pushed a commit to Raahul46/FBGEMM-1 that referenced this pull request Jun 3, 2025
Summary:

X-link: facebookresearch/FBGEMM#1288

Design doc: https://docs.google.com/document/d/149LdAEHOLP7ei4hwVVkAFXGa4N9uLs1J7efxfBZp3dY/edit?tab=t.0#heading=h.49t3yfaqmt54

Context:
We are enabling the usage of rocksDB checkpoint feature in KVTensorWrapper. This allows us to create checkpoints of the embedding tables in SSD. Later, these checkpoints are used by the checkpointing component to create a checkpoint and upload it it to the manifold 

In this diff:

We are adding all the helper functions to create folder structures in SSDs for storing the rocksDB checkpoints

Reviewed By: duduyi2013

Differential Revision: D75489819
@Raahul46 Raahul46 force-pushed the export-D75489819 branch from d598938 to 57224f7 Compare June 3, 2025 17:03
Raahul46 pushed a commit to Raahul46/FBGEMM-1 that referenced this pull request Jun 3, 2025
Summary:

X-link: facebookresearch/FBGEMM#1288

Design doc: https://docs.google.com/document/d/149LdAEHOLP7ei4hwVVkAFXGa4N9uLs1J7efxfBZp3dY/edit?tab=t.0#heading=h.49t3yfaqmt54

Context:
We are enabling the usage of rocksDB checkpoint feature in KVTensorWrapper. This allows us to create checkpoints of the embedding tables in SSD. Later, these checkpoints are used by the checkpointing component to create a checkpoint and upload it it to the manifold 

In this diff:

We are adding all the helper functions to create folder structures in SSDs for storing the rocksDB checkpoints

Reviewed By: duduyi2013

Differential Revision: D75489819
Raahul46 pushed a commit to Raahul46/FBGEMM-1 that referenced this pull request Jun 3, 2025
Summary:

X-link: facebookresearch/FBGEMM#1288

Design doc: https://docs.google.com/document/d/149LdAEHOLP7ei4hwVVkAFXGa4N9uLs1J7efxfBZp3dY/edit?tab=t.0#heading=h.49t3yfaqmt54

Context:
We are enabling the usage of rocksDB checkpoint feature in KVTensorWrapper. This allows us to create checkpoints of the embedding tables in SSD. Later, these checkpoints are used by the checkpointing component to create a checkpoint and upload it it to the manifold 

In this diff:

We are adding all the helper functions to create folder structures in SSDs for storing the rocksDB checkpoints

Reviewed By: duduyi2013

Differential Revision: D75489819
@Raahul46 Raahul46 force-pushed the export-D75489819 branch from 57224f7 to e153227 Compare June 3, 2025 17:04
Raahul46 pushed a commit to Raahul46/FBGEMM-1 that referenced this pull request Jun 3, 2025
Summary:

X-link: facebookresearch/FBGEMM#1288

Design doc: https://docs.google.com/document/d/149LdAEHOLP7ei4hwVVkAFXGa4N9uLs1J7efxfBZp3dY/edit?tab=t.0#heading=h.49t3yfaqmt54

Context:
We are enabling the usage of rocksDB checkpoint feature in KVTensorWrapper. This allows us to create checkpoints of the embedding tables in SSD. Later, these checkpoints are used by the checkpointing component to create a checkpoint and upload it it to the manifold 

In this diff:

We are adding all the helper functions to create folder structures in SSDs for storing the rocksDB checkpoints

Reviewed By: duduyi2013

Differential Revision: D75489819
@facebook-github-bot
Copy link
Contributor

This pull request was exported from Phabricator. Differential Revision: D75489819

@Raahul46 Raahul46 force-pushed the export-D75489819 branch from e153227 to 2371fad Compare June 3, 2025 17:10
Raahul46 pushed a commit to Raahul46/FBGEMM-1 that referenced this pull request Jun 3, 2025
Summary:
Pull Request resolved: pytorch#4213

X-link: facebookresearch/FBGEMM#1288

Design doc: https://docs.google.com/document/d/149LdAEHOLP7ei4hwVVkAFXGa4N9uLs1J7efxfBZp3dY/edit?tab=t.0#heading=h.49t3yfaqmt54

Context:
We are enabling the usage of rocksDB checkpoint feature in KVTensorWrapper. This allows us to create checkpoints of the embedding tables in SSD. Later, these checkpoints are used by the checkpointing component to create a checkpoint and upload it it to the manifold

In this diff:

We are adding all the helper functions to create folder structures in SSDs for storing the rocksDB checkpoints

Reviewed By: duduyi2013

Differential Revision: D75489819
@facebook-github-bot
Copy link
Contributor

This pull request was exported from Phabricator. Differential Revision: D75489819

@Raahul46 Raahul46 force-pushed the export-D75489819 branch from 2371fad to 0c311c8 Compare June 3, 2025 17:17
Raahul46 pushed a commit to Raahul46/FBGEMM-1 that referenced this pull request Jun 3, 2025
Summary:
Pull Request resolved: pytorch#4213

X-link: facebookresearch/FBGEMM#1288

Design doc: https://docs.google.com/document/d/149LdAEHOLP7ei4hwVVkAFXGa4N9uLs1J7efxfBZp3dY/edit?tab=t.0#heading=h.49t3yfaqmt54

Context:
We are enabling the usage of rocksDB checkpoint feature in KVTensorWrapper. This allows us to create checkpoints of the embedding tables in SSD. Later, these checkpoints are used by the checkpointing component to create a checkpoint and upload it it to the manifold

In this diff:

We are adding all the helper functions to create folder structures in SSDs for storing the rocksDB checkpoints

Reviewed By: duduyi2013

Differential Revision: D75489819
@Raahul46 Raahul46 force-pushed the export-D75489819 branch from 0c311c8 to 54a112d Compare June 4, 2025 05:25
Raahul46 pushed a commit to Raahul46/FBGEMM-1 that referenced this pull request Jun 4, 2025
Summary:

X-link: facebookresearch/FBGEMM#1288

Design doc: https://docs.google.com/document/d/149LdAEHOLP7ei4hwVVkAFXGa4N9uLs1J7efxfBZp3dY/edit?tab=t.0#heading=h.49t3yfaqmt54

Context:
We are enabling the usage of rocksDB checkpoint feature in KVTensorWrapper. This allows us to create checkpoints of the embedding tables in SSD. Later, these checkpoints are used by the checkpointing component to create a checkpoint and upload it it to the manifold 

In this diff:

We are adding all the helper functions to create folder structures in SSDs for storing the rocksDB checkpoints

Reviewed By: duduyi2013

Differential Revision: D75489819
Raahul46 pushed a commit to Raahul46/FBGEMM-1 that referenced this pull request Jun 4, 2025
Summary:

X-link: facebookresearch/FBGEMM#1288

Design doc: https://docs.google.com/document/d/149LdAEHOLP7ei4hwVVkAFXGa4N9uLs1J7efxfBZp3dY/edit?tab=t.0#heading=h.49t3yfaqmt54

Context:
We are enabling the usage of rocksDB checkpoint feature in KVTensorWrapper. This allows us to create checkpoints of the embedding tables in SSD. Later, these checkpoints are used by the checkpointing component to create a checkpoint and upload it it to the manifold 

In this diff:

We are adding all the helper functions to create folder structures in SSDs for storing the rocksDB checkpoints

Reviewed By: duduyi2013

Differential Revision: D75489819
@Raahul46 Raahul46 force-pushed the export-D75489819 branch from 54a112d to fb3d558 Compare June 4, 2025 05:26
@facebook-github-bot
Copy link
Contributor

This pull request was exported from Phabricator. Differential Revision: D75489819

Raahul46 pushed a commit to Raahul46/FBGEMM-1 that referenced this pull request Jun 4, 2025
Summary:
Pull Request resolved: pytorch#4213

X-link: facebookresearch/FBGEMM#1288

Design doc: https://docs.google.com/document/d/149LdAEHOLP7ei4hwVVkAFXGa4N9uLs1J7efxfBZp3dY/edit?tab=t.0#heading=h.49t3yfaqmt54

Context:
We are enabling the usage of rocksDB checkpoint feature in KVTensorWrapper. This allows us to create checkpoints of the embedding tables in SSD. Later, these checkpoints are used by the checkpointing component to create a checkpoint and upload it it to the manifold

In this diff:

We are adding all the helper functions to create folder structures in SSDs for storing the rocksDB checkpoints

Reviewed By: duduyi2013

Differential Revision: D75489819
Raahul46 pushed a commit to Raahul46/FBGEMM-1 that referenced this pull request Jun 5, 2025
Summary:

X-link: facebookresearch/FBGEMM#1288

Design doc: https://docs.google.com/document/d/149LdAEHOLP7ei4hwVVkAFXGa4N9uLs1J7efxfBZp3dY/edit?tab=t.0#heading=h.49t3yfaqmt54

Context:
We are enabling the usage of rocksDB checkpoint feature in KVTensorWrapper. This allows us to create checkpoints of the embedding tables in SSD. Later, these checkpoints are used by the checkpointing component to create a checkpoint and upload it it to the manifold 

In this diff:

We are adding all the helper functions to create folder structures in SSDs for storing the rocksDB checkpoints

Reviewed By: duduyi2013

Differential Revision: D75489819
Raahul46 pushed a commit to Raahul46/FBGEMM-1 that referenced this pull request Jun 5, 2025
Summary:

X-link: facebookresearch/FBGEMM#1288

Design doc: https://docs.google.com/document/d/149LdAEHOLP7ei4hwVVkAFXGa4N9uLs1J7efxfBZp3dY/edit?tab=t.0#heading=h.49t3yfaqmt54

Context:
We are enabling the usage of rocksDB checkpoint feature in KVTensorWrapper. This allows us to create checkpoints of the embedding tables in SSD. Later, these checkpoints are used by the checkpointing component to create a checkpoint and upload it it to the manifold 

In this diff:

We are adding all the helper functions to create folder structures in SSDs for storing the rocksDB checkpoints

Reviewed By: duduyi2013

Differential Revision: D75489819
Raahul46 pushed a commit to Raahul46/FBGEMM-1 that referenced this pull request Jun 5, 2025
Summary:

X-link: facebookresearch/FBGEMM#1288

Design doc: https://docs.google.com/document/d/149LdAEHOLP7ei4hwVVkAFXGa4N9uLs1J7efxfBZp3dY/edit?tab=t.0#heading=h.49t3yfaqmt54

Context:
We are enabling the usage of rocksDB checkpoint feature in KVTensorWrapper. This allows us to create checkpoints of the embedding tables in SSD. Later, these checkpoints are used by the checkpointing component to create a checkpoint and upload it it to the manifold 

In this diff:

We are adding all the helper functions to create folder structures in SSDs for storing the rocksDB checkpoints

Reviewed By: duduyi2013

Differential Revision: D75489819
Raahul46 pushed a commit to Raahul46/FBGEMM-1 that referenced this pull request Jun 5, 2025
Summary:

X-link: facebookresearch/FBGEMM#1288

Design doc: https://docs.google.com/document/d/149LdAEHOLP7ei4hwVVkAFXGa4N9uLs1J7efxfBZp3dY/edit?tab=t.0#heading=h.49t3yfaqmt54

Context:
We are enabling the usage of rocksDB checkpoint feature in KVTensorWrapper. This allows us to create checkpoints of the embedding tables in SSD. Later, these checkpoints are used by the checkpointing component to create a checkpoint and upload it it to the manifold 

In this diff:

We are adding all the helper functions to create folder structures in SSDs for storing the rocksDB checkpoints

Reviewed By: duduyi2013

Differential Revision: D75489819
Raahul46 pushed a commit to Raahul46/FBGEMM-1 that referenced this pull request Jun 5, 2025
Summary:

X-link: facebookresearch/FBGEMM#1288

Design doc: https://docs.google.com/document/d/149LdAEHOLP7ei4hwVVkAFXGa4N9uLs1J7efxfBZp3dY/edit?tab=t.0#heading=h.49t3yfaqmt54

Context:
We are enabling the usage of rocksDB checkpoint feature in KVTensorWrapper. This allows us to create checkpoints of the embedding tables in SSD. Later, these checkpoints are used by the checkpointing component to create a checkpoint and upload it it to the manifold 

In this diff:

We are adding all the helper functions to create folder structures in SSDs for storing the rocksDB checkpoints

Reviewed By: duduyi2013

Differential Revision: D75489819
@Raahul46 Raahul46 force-pushed the export-D75489819 branch from aa722b2 to e151b41 Compare June 5, 2025 02:23
@facebook-github-bot
Copy link
Contributor

This pull request was exported from Phabricator. Differential Revision: D75489819

@Raahul46 Raahul46 force-pushed the export-D75489819 branch from e151b41 to 6f69dc1 Compare June 5, 2025 17:59
Raahul46 pushed a commit to Raahul46/FBGEMM-1 that referenced this pull request Jun 5, 2025
Summary:

X-link: facebookresearch/FBGEMM#1288

Design doc: https://docs.google.com/document/d/149LdAEHOLP7ei4hwVVkAFXGa4N9uLs1J7efxfBZp3dY/edit?tab=t.0#heading=h.49t3yfaqmt54

Context:
We are enabling the usage of rocksDB checkpoint feature in KVTensorWrapper. This allows us to create checkpoints of the embedding tables in SSD. Later, these checkpoints are used by the checkpointing component to create a checkpoint and upload it it to the manifold 

In this diff:

We are adding all the helper functions to create folder structures in SSDs for storing the rocksDB checkpoints

Reviewed By: duduyi2013

Differential Revision: D75489819
Raahul46 pushed a commit to Raahul46/FBGEMM-1 that referenced this pull request Jun 5, 2025
Summary:

X-link: facebookresearch/FBGEMM#1288

Design doc: https://docs.google.com/document/d/149LdAEHOLP7ei4hwVVkAFXGa4N9uLs1J7efxfBZp3dY/edit?tab=t.0#heading=h.49t3yfaqmt54

Context:
We are enabling the usage of rocksDB checkpoint feature in KVTensorWrapper. This allows us to create checkpoints of the embedding tables in SSD. Later, these checkpoints are used by the checkpointing component to create a checkpoint and upload it it to the manifold 

In this diff:

We are adding all the helper functions to create folder structures in SSDs for storing the rocksDB checkpoints

Reviewed By: duduyi2013

Differential Revision: D75489819
Raahul46 pushed a commit to Raahul46/FBGEMM-1 that referenced this pull request Jun 5, 2025
Summary:

X-link: facebookresearch/FBGEMM#1288

Design doc: https://docs.google.com/document/d/149LdAEHOLP7ei4hwVVkAFXGa4N9uLs1J7efxfBZp3dY/edit?tab=t.0#heading=h.49t3yfaqmt54

Context:
We are enabling the usage of rocksDB checkpoint feature in KVTensorWrapper. This allows us to create checkpoints of the embedding tables in SSD. Later, these checkpoints are used by the checkpointing component to create a checkpoint and upload it it to the manifold 

In this diff:

We are adding all the helper functions to create folder structures in SSDs for storing the rocksDB checkpoints

Reviewed By: duduyi2013

Differential Revision: D75489819
Raahul46 pushed a commit to Raahul46/FBGEMM-1 that referenced this pull request Jun 5, 2025
Summary:

X-link: facebookresearch/FBGEMM#1288

Design doc: https://docs.google.com/document/d/149LdAEHOLP7ei4hwVVkAFXGa4N9uLs1J7efxfBZp3dY/edit?tab=t.0#heading=h.49t3yfaqmt54

Context:
We are enabling the usage of rocksDB checkpoint feature in KVTensorWrapper. This allows us to create checkpoints of the embedding tables in SSD. Later, these checkpoints are used by the checkpointing component to create a checkpoint and upload it it to the manifold 

In this diff:

We are adding all the helper functions to create folder structures in SSDs for storing the rocksDB checkpoints

Reviewed By: duduyi2013

Differential Revision: D75489819
Raahul46 pushed a commit to Raahul46/FBGEMM-1 that referenced this pull request Jun 5, 2025
Summary:

X-link: facebookresearch/FBGEMM#1288

Design doc: https://docs.google.com/document/d/149LdAEHOLP7ei4hwVVkAFXGa4N9uLs1J7efxfBZp3dY/edit?tab=t.0#heading=h.49t3yfaqmt54

Context:
We are enabling the usage of rocksDB checkpoint feature in KVTensorWrapper. This allows us to create checkpoints of the embedding tables in SSD. Later, these checkpoints are used by the checkpointing component to create a checkpoint and upload it it to the manifold 

In this diff:

We are adding all the helper functions to create folder structures in SSDs for storing the rocksDB checkpoints

Reviewed By: duduyi2013

Differential Revision: D75489819
@Raahul46 Raahul46 force-pushed the export-D75489819 branch from 6f69dc1 to 4a67d9b Compare June 5, 2025 18:00
Raahul46 pushed a commit to Raahul46/FBGEMM-1 that referenced this pull request Jun 5, 2025
Summary:

X-link: facebookresearch/FBGEMM#1288

Design doc: https://docs.google.com/document/d/149LdAEHOLP7ei4hwVVkAFXGa4N9uLs1J7efxfBZp3dY/edit?tab=t.0#heading=h.49t3yfaqmt54

Context:
We are enabling the usage of rocksDB checkpoint feature in KVTensorWrapper. This allows us to create checkpoints of the embedding tables in SSD. Later, these checkpoints are used by the checkpointing component to create a checkpoint and upload it it to the manifold 

In this diff:

We are adding all the helper functions to create folder structures in SSDs for storing the rocksDB checkpoints

Reviewed By: duduyi2013

Differential Revision: D75489819
Raahul46 pushed a commit to Raahul46/FBGEMM-1 that referenced this pull request Jun 5, 2025
Summary:

X-link: facebookresearch/FBGEMM#1288

Design doc: https://docs.google.com/document/d/149LdAEHOLP7ei4hwVVkAFXGa4N9uLs1J7efxfBZp3dY/edit?tab=t.0#heading=h.49t3yfaqmt54

Context:
We are enabling the usage of rocksDB checkpoint feature in KVTensorWrapper. This allows us to create checkpoints of the embedding tables in SSD. Later, these checkpoints are used by the checkpointing component to create a checkpoint and upload it it to the manifold 

In this diff:

We are adding all the helper functions to create folder structures in SSDs for storing the rocksDB checkpoints

Reviewed By: duduyi2013

Differential Revision: D75489819
Raahul46 pushed a commit to Raahul46/FBGEMM-1 that referenced this pull request Jun 5, 2025
Summary:

X-link: facebookresearch/FBGEMM#1288

Design doc: https://docs.google.com/document/d/149LdAEHOLP7ei4hwVVkAFXGa4N9uLs1J7efxfBZp3dY/edit?tab=t.0#heading=h.49t3yfaqmt54

Context:
We are enabling the usage of rocksDB checkpoint feature in KVTensorWrapper. This allows us to create checkpoints of the embedding tables in SSD. Later, these checkpoints are used by the checkpointing component to create a checkpoint and upload it it to the manifold 

In this diff:

We are adding all the helper functions to create folder structures in SSDs for storing the rocksDB checkpoints

Reviewed By: duduyi2013

Differential Revision: D75489819
Raahul46 pushed a commit to Raahul46/FBGEMM-1 that referenced this pull request Jun 5, 2025
Summary:

X-link: facebookresearch/FBGEMM#1288

Design doc: https://docs.google.com/document/d/149LdAEHOLP7ei4hwVVkAFXGa4N9uLs1J7efxfBZp3dY/edit?tab=t.0#heading=h.49t3yfaqmt54

Context:
We are enabling the usage of rocksDB checkpoint feature in KVTensorWrapper. This allows us to create checkpoints of the embedding tables in SSD. Later, these checkpoints are used by the checkpointing component to create a checkpoint and upload it it to the manifold 

In this diff:

We are adding all the helper functions to create folder structures in SSDs for storing the rocksDB checkpoints

Reviewed By: duduyi2013

Differential Revision: D75489819
Raahul46 pushed a commit to Raahul46/FBGEMM-1 that referenced this pull request Jun 5, 2025
Summary:

X-link: facebookresearch/FBGEMM#1288

Design doc: https://docs.google.com/document/d/149LdAEHOLP7ei4hwVVkAFXGa4N9uLs1J7efxfBZp3dY/edit?tab=t.0#heading=h.49t3yfaqmt54

Context:
We are enabling the usage of rocksDB checkpoint feature in KVTensorWrapper. This allows us to create checkpoints of the embedding tables in SSD. Later, these checkpoints are used by the checkpointing component to create a checkpoint and upload it it to the manifold 

In this diff:

We are adding all the helper functions to create folder structures in SSDs for storing the rocksDB checkpoints

Reviewed By: duduyi2013

Differential Revision: D75489819
Raahul46 pushed a commit to Raahul46/FBGEMM-1 that referenced this pull request Jun 5, 2025
Summary:

X-link: facebookresearch/FBGEMM#1288

Design doc: https://docs.google.com/document/d/149LdAEHOLP7ei4hwVVkAFXGa4N9uLs1J7efxfBZp3dY/edit?tab=t.0#heading=h.49t3yfaqmt54

Context:
We are enabling the usage of rocksDB checkpoint feature in KVTensorWrapper. This allows us to create checkpoints of the embedding tables in SSD. Later, these checkpoints are used by the checkpointing component to create a checkpoint and upload it it to the manifold 

In this diff:

We are adding all the helper functions to create folder structures in SSDs for storing the rocksDB checkpoints

Reviewed By: duduyi2013

Differential Revision: D75489819
@facebook-github-bot
Copy link
Contributor

This pull request was exported from Phabricator. Differential Revision: D75489819

@Raahul46 Raahul46 force-pushed the export-D75489819 branch from 4a67d9b to 6711ce8 Compare June 5, 2025 18:02
Raahul46 pushed a commit to Raahul46/FBGEMM-1 that referenced this pull request Jun 5, 2025
Summary:
Pull Request resolved: pytorch#4213

X-link: facebookresearch/FBGEMM#1288

Design doc: https://docs.google.com/document/d/149LdAEHOLP7ei4hwVVkAFXGa4N9uLs1J7efxfBZp3dY/edit?tab=t.0#heading=h.49t3yfaqmt54

Context:
We are enabling the usage of rocksDB checkpoint feature in KVTensorWrapper. This allows us to create checkpoints of the embedding tables in SSD. Later, these checkpoints are used by the checkpointing component to create a checkpoint and upload it it to the manifold

In this diff:

We are adding all the helper functions to create folder structures in SSDs for storing the rocksDB checkpoints

Reviewed By: duduyi2013

Differential Revision: D75489819
Summary:
Pull Request resolved: pytorch#4213

X-link: facebookresearch/FBGEMM#1288

Design doc: https://docs.google.com/document/d/149LdAEHOLP7ei4hwVVkAFXGa4N9uLs1J7efxfBZp3dY/edit?tab=t.0#heading=h.49t3yfaqmt54

Context:
We are enabling the usage of rocksDB checkpoint feature in KVTensorWrapper. This allows us to create checkpoints of the embedding tables in SSD. Later, these checkpoints are used by the checkpointing component to create a checkpoint and upload it it to the manifold

In this diff:

We are adding all the helper functions to create folder structures in SSDs for storing the rocksDB checkpoints

Reviewed By: duduyi2013

Differential Revision: D75489819
@facebook-github-bot
Copy link
Contributor

This pull request was exported from Phabricator. Differential Revision: D75489819

@facebook-github-bot
Copy link
Contributor

This pull request has been merged in 95f8e76.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants