From 94f7d99457bd2870a25ce27df4704a3bc8459a34 Mon Sep 17 00:00:00 2001
From: FindHao <find@findhao.net>
Date: Fri, 28 Apr 2023 09:44:28 -0400
Subject: [PATCH 1/3] update citation

---
 README.md | 14 ++++++++++++++
 1 file changed, 14 insertions(+)

diff --git a/README.md b/README.md
index 220aa4e13f..f8427e1311 100644
--- a/README.md
+++ b/README.md
@@ -145,3 +145,17 @@ See [Unidash](https://www.internalfb.com/intern/unidash/dashboard/pytorch_benchm
 ## Adding new models
 
 See [Adding Models](torchbenchmark/models/ADDING_MODELS.md).
+
+## Citation
+
+If you use TorchBench in your academic work, please cite the following:
+```
+@misc{hao2023torchbench,
+      title={TorchBench: Benchmarking PyTorch with High API Surface Coverage}, 
+      author={Yueming Hao and Xu Zhao and Bin Bao and David Berard and Will Constable and Adnan Aziz and Xu Liu},
+      year={2023},
+      eprint={2304.14226},
+      archivePrefix={arXiv},
+      primaryClass={cs.LG}
+}
+```
\ No newline at end of file

From 027ebe1275e2166dd6e323ed8bfef6a5011b2fff Mon Sep 17 00:00:00 2001
From: FindHao <find@findhao.net>
Date: Fri, 28 Apr 2023 09:50:09 -0400
Subject: [PATCH 2/3] update citation.cff

---
 CITATION.cff | 22 ++++++++++++++++++++++
 1 file changed, 22 insertions(+)

diff --git a/CITATION.cff b/CITATION.cff
index bfcec8ad24..d73232745b 100644
--- a/CITATION.cff
+++ b/CITATION.cff
@@ -26,3 +26,25 @@ authors:
 message: "If you use this software, please cite it using these metadata."
 date-released: 2020-09-03
 repository-code: "https://github.com/pytorch/benchmark"
+preferred-citation:
+  type: article
+  authors:
+  - family-names: "Hao"
+    given-names: "Yueming"
+  - family-names: "Zhao"
+    given-names: "Xu"
+  - family-names: "Bao"
+    given-names: "Bin"
+  - family-names: "Berard"
+    given-names: "David"
+  - family-names: "Constable"
+    given-names: "Will"
+  - family-names: "Aziz"
+    given-names: "Adnan"
+  - family-names: "Liu"
+    given-names: "Xu"
+  title: "TorchBench: Benchmarking PyTorch with High API Surface Coverage"
+  year: 2023
+  eprint: "2304.14226"
+  primary-class: "cs.LG"
+  archivePrefix: "arXiv"  
\ No newline at end of file

From fd844b1efd4190c70ba0e7bf6e88bef3154e4ac9 Mon Sep 17 00:00:00 2001
From: FindHao <find@findhao.net>
Date: Fri, 28 Apr 2023 13:01:00 -0400
Subject: [PATCH 3/3] add arxiv link

---
 README.md | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/README.md b/README.md
index f8427e1311..e5cefef3b7 100644
--- a/README.md
+++ b/README.md
@@ -158,4 +158,5 @@ If you use TorchBench in your academic work, please cite the following:
       archivePrefix={arXiv},
       primaryClass={cs.LG}
 }
-```
\ No newline at end of file
+```
+The full paper [TorchBench: Benchmarking PyTorch with High API Surface Coverage](https://arxiv.org/abs/2304.14226) is public on arXiv.org.