From 5522a8bdb41b85a7d3d0c50a2322d96697e5b0c3 Mon Sep 17 00:00:00 2001
From: Philippe Tillet <ptillet@g.harvard.edu>
Date: Fri, 5 Feb 2021 11:31:58 -0800
Subject: [PATCH] [GENERAL] Added continuous integration

---
 .ci/azure-pipelines.yml | 47 +++++++++++++++++++++++++++++++++++++++++
 README.md               |  2 ++
 2 files changed, 49 insertions(+)
 create mode 100644 .ci/azure-pipelines.yml

diff --git a/.ci/azure-pipelines.yml b/.ci/azure-pipelines.yml
new file mode 100644
index 000000000..e78aa0bc7
--- /dev/null
+++ b/.ci/azure-pipelines.yml
@@ -0,0 +1,47 @@
+# Python package
+# Create and test a Python package on multiple Python versions.
+# Add steps that analyze code, save the dist with the build record, publish to a PyPI-compatible index, and more:
+# https://docs.microsoft.com/azure/devops/pipelines/languages/python
+
+trigger:
+- master
+
+pool:
+  name: default
+
+variables:
+- name: venv
+  value: venv
+
+strategy:
+  matrix:
+    Python37:
+      python.version: '3.7'
+  maxParallel: 1
+
+steps:
+- task: UsePythonVersion@0
+  inputs:
+    versionSpec: '$(python.version)'
+    architecture: 'x64'
+  displayName: 'Use Python $(python.version)'
+
+- script: |
+    mkdir $(venv)
+    python -m virtualenv $(venv)
+    source $(venv)/bin/activate
+  displayName: setup virtual environment
+
+- script: |
+    python -m pip install --upgrade pip
+    pip install torch
+  displayName: 'Install dependencies'
+
+- script: |
+    cd python
+    python setup.py install
+  displayName: 'Install Triton'
+
+- script: |
+    pytest .
+  displayName: 'Python Unit Tests'
\ No newline at end of file
diff --git a/README.md b/README.md
index f2b27b8d3..65cf14607 100644
--- a/README.md
+++ b/README.md
@@ -2,6 +2,8 @@
 
 This is the development repository of Triton, a language and compiler for writing highly efficient custom Deep-Learning primitives. The aim of Triton is to provide an open-source environment to write fast code at higher productivity than CUDA, but also with higher flexibility than other existing DSLs.
 
+[![Build Status](https://dev.azure.com/phil0342/Triton/_apis/build/status/ptillet.triton?branchName=refs%2Fpull%2F58%2Fmerge)](https://dev.azure.com/phil0342/Triton/_build/latest?definitionId=6&branchName=refs%2Fpull%2F58%2Fmerge)
+
 The foundations of this project are described in the following MAPL2019 publication: [Triton: An Intermediate Language and Compiler for Tiled Neural Network Computations](http://www.eecs.harvard.edu/~htk/publication/2019-mapl-tillet-kung-cox.pdf). Please consider citing us if you use our work!